add back importand val_dump step

wandb · Dec 12, 2024 · c30ef52 · c30ef52
1 parent f3e513b
commit c30ef52
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/weave/trace_server/clickhouse_trace_server_batched.py b/weave/trace_server/clickhouse_trace_server_batched.py
@@ -1869,6 +1869,11 @@ def _select_objs_query(
         for row in query_result:
             (object_id, digest, val_dump) = row
             object_values[(object_id, digest)] = val_dump
+
+        # update the val_dump for each object
+        for obj in result:
+            obj.val_dump = object_values.get((obj.object_id, obj.digest), "{}")
+
         return result
 
     def _run_migrations(self) -> None: