small adjustment

wandb · Jan 6, 2025 · c6addc7 · c6addc7
1 parent ad17e17
commit c6addc7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/weave/flow/eval.py b/weave/flow/eval.py
@@ -270,7 +270,11 @@ async def predict_and_score(self, model: Union[Op, Model], example: dict) -> dic
         except Exception:
             print("model_output failed")
             traceback.print_exc()
-            model_output = None
+            return {
+                self._output_key: None,
+                "scores": {},
+                "model_latency": time.time() - model_start_time,
+            }
         model_latency = time.time() - model_start_time
 
         scores = {}  # TODO: Consider moving scorer setup and checks out of `predict_and_score`