@@ -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