diff --git a/weave/flow/eval.py b/weave/flow/eval.py index 1f71f58bb262..4e93b304b731 100644 --- 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`