lint

wandb · Jan 7, 2025 · e0a5017 · e0a5017
1 parent f46fa06
commit e0a5017
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/weave/trace/weave_client.py b/weave/trace/weave_client.py
@@ -469,7 +469,10 @@ async def apply_scorer(
 
         model_inputs = {k: v for k, v in self.inputs.items() if k != "self"}
         example = {**model_inputs, **(additional_scorer_kwargs or {})}
-        apply_scorer_result = await apply_scorer_async(scorer, example, self.output)
+        output = self.output
+        if isinstance(output, Ref):
+            output = output.get()
+        apply_scorer_result = await apply_scorer_async(scorer, example, output)
         score_call = apply_scorer_result.score_call
 
         wc = weave_client_context.get_weave_client()
@@ -866,7 +869,7 @@ def finish_call(
             postprocessed_output = original_output
         self._save_nested_objects(postprocessed_output)
 
-        call.output = map_to_refs(postprocessed_output)
+        # call.output = map_to_refs(postprocessed_output)
 
         # Summary handling
         summary = {}