undo

wandb · Dec 13, 2024 · 60133fd · 60133fd
1 parent 23326e1
commit 60133fd
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/weave/trace_server/objects_query_builder.py b/weave/trace_server/objects_query_builder.py
@@ -176,11 +176,10 @@ def add_is_op_condition(self, is_op: bool) -> None:
             self._conditions.append("is_op = 0")
 
     def add_base_object_classes_condition(self, base_object_classes: list[str]) -> None:
-        param_key = "base_object_classes"
         self._conditions.append(
-            f"base_object_classes IN {{{param_key}: Array(String)}}"
+            "base_object_class IN {base_object_classes: Array(String)}"
         )
-        self.parameters.update({param_key: base_object_classes})
+        self.parameters.update({"base_object_classes": base_object_classes})
 
     def add_order(self, field: str, direction: str) -> None:
         direction = direction.lower()