lint pass 4

wandb · Dec 12, 2024 · fc97858 · fc97858
1 parent 39e8f46
commit fc97858
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/trace/builtin_objects/test_builtin_scorer.py b/tests/trace/builtin_objects/test_builtin_scorer.py
@@ -6,7 +6,7 @@
 # 5. Remote Create, Remote Direct Score
 import weave
 from weave.builtin_objects.scorers.LLMJudgeScorer import LLMJudgeScorer
-from weave.trace.weave_client import ApplyScorerResult, Call, WeaveClient
+from weave.trace.weave_client import Call, WeaveClient
 from weave.trace_server import trace_server_interface as tsi
 
 scorer_args = {
@@ -64,7 +64,9 @@ def simple_op(question: str) -> str:
 
 
 def assert_expected_outcome(
-    target_call: Call, scorer_res: ApplyScorerResult | tsi.ScoreCallRes
+    # target_call: Call, scorer_res: ApplyScorerResult | tsi.ScoreCallRes
+    target_call: Call,
+    scorer_res: dict | tsi.ScoreCallRes,
 ):
     scorer_output = None
     feedback_id = None