Skip to content

Commit 50f7deb

Browse files
BarrySlyDelgadoBarry Sly-Delgado
andauthored
Remove prune_files option (#4234)
* add doc for prune depth * lint --------- Co-authored-by: Barry Sly-Delgado <[email protected]>
1 parent 44a9eb6 commit 50f7deb

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

taskvine/src/bindings/python3/ndcctools/taskvine/dask_executor.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ class DaskVine(Manager):
121121
# fn(*args) at some point during its execution to produce the dask task result.
122122
# Should return a tuple of (wrapper result, dask call result). Use for debugging.
123123
# @param wrapper_proc Function to process results from wrapper on completion. (default is print)
124-
# @param prune_files If True, remove files from the cluster after they are no longer needed.
124+
# @param prune_depth Control pruning behavior: 0 (default) - no pruning, 1 - only check direct consumers, 2+ - check consumers up to specified depth
125125
def get(self, dsk, keys, *,
126126
environment=None,
127127
extra_files=None,
@@ -147,7 +147,6 @@ def get(self, dsk, keys, *,
147147
progress_label="[green]tasks",
148148
wrapper=None,
149149
wrapper_proc=print,
150-
prune_files=False,
151150
prune_depth=0,
152151
hoisting_modules=None, # Deprecated, use lib_modules
153152
import_modules=None, # Deprecated, use lib_modules
@@ -196,7 +195,6 @@ def get(self, dsk, keys, *,
196195
self.wrapper = wrapper
197196
self.wrapper_proc = wrapper_proc
198197
self.prune_depth = prune_depth
199-
self.prune_files = prune_files
200198
self.category_info = defaultdict(lambda: {"num_tasks": 0, "total_execution_time": 0})
201199
self.max_priority = float('inf')
202200
self.min_priority = float('-inf')

0 commit comments

Comments
 (0)