Update daft/runners/pyrunner.py

Eventual-Inc · Jul 30, 2024 · 5051980 · 5051980
1 parent 50cc5e2
commit 5051980
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/daft/runners/pyrunner.py b/daft/runners/pyrunner.py
@@ -382,7 +382,7 @@ def build_partitions(
             for part, partial in zip(partitions, final_metadata)
         ]
 
-        # Release CPU and GPU resources, but not memory since the future has not been consumed yet
+        # Release CPU, GPU and memory resources
         with self._resource_accounting_lock:
             self._available_bytes_memory += resource_request.memory_bytes or 0
             self._available_cpus += resource_request.num_cpus or 0.0