Fix PartitionTaskBuilder as well

Eventual-Inc · Oct 17, 2023 · 0a48456 · 0a48456
1 parent ac48c5f
commit 0a48456
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/daft/execution/execution_step.py b/daft/execution/execution_step.py
@@ -150,7 +150,7 @@ def finalize_partition_task_multi_output(self) -> MultiOutputPartitionTask[Parti
     def __str__(self) -> str:
         return (
             f"PartitionTaskBuilder\n"
-            f"  Inputs: {self.inputs}\n"
+            f"  Inputs: <{len(self.inputs)} partitions>\n"
             f"  Resource Request: {self.resource_request}\n"
             f"  Instructions: {[i.__class__.__name__ for i in self.instructions]}"
         )