Remove unused array_names parameter in executors

cubed-dev · Feb 14, 2024 · c062158 · c062158
1 parent a19cfad
commit c062158
Show file tree

Hide file tree

Showing 10 changed files with 1 addition and 23 deletions.
diff --git a/cubed/core/array.py b/cubed/core/array.py
@@ -286,7 +286,6 @@ def compute(
         optimize_graph=optimize_graph,
         optimize_function=optimize_function,
         resume=resume,
-        array_names=[a.name for a in arrays],
         spec=spec,
         **kwargs,
     )

diff --git a/cubed/core/plan.py b/cubed/core/plan.py
@@ -193,7 +193,6 @@ def execute(
         optimize_function=None,
         resume=None,
         spec=None,
-        array_names=None,
         **kwargs,
     ):
         dag = self._finalize_dag(optimize_graph, optimize_function)
@@ -204,7 +203,6 @@ def execute(
         executor.execute_dag(
             dag,
             callbacks=callbacks,
-            array_names=array_names,
             resume=resume,
             spec=spec,
             **kwargs,

diff --git a/cubed/runtime/executors/beam.py b/cubed/runtime/executors/beam.py
@@ -82,9 +82,7 @@ def expand(self, pcoll):
 class BeamDagExecutor(DagExecutor):
     """An execution engine that uses Apache Beam."""
 
-    def execute_dag(
-        self, dag, callbacks=None, array_names=None, resume=None, spec=None, **kwargs
-    ):
+    def execute_dag(self, dag, callbacks=None, resume=None, spec=None, **kwargs):
         dag = dag.copy()
         pipeline = beam.Pipeline(**kwargs)
 

diff --git a/cubed/runtime/executors/coiled.py b/cubed/runtime/executors/coiled.py
@@ -20,7 +20,6 @@ def execute_dag(
         self,
         dag: MultiDiGraph,
         callbacks: Optional[Sequence[Callback]] = None,
-        array_names: Optional[Sequence[str]] = None,
         resume: Optional[bool] = None,
         spec: Optional[Spec] = None,
         **coiled_kwargs: Mapping[str, Any],

diff --git a/cubed/runtime/executors/dask_distributed_async.py b/cubed/runtime/executors/dask_distributed_async.py
@@ -110,7 +110,6 @@ def check_runtime_memory(spec, client):
 async def async_execute_dag(
     dag: MultiDiGraph,
     callbacks: Optional[Sequence[Callback]] = None,
-    array_names: Optional[Sequence[str]] = None,
     resume: Optional[bool] = None,
     spec: Optional[Spec] = None,
     compute_arrays_in_parallel: Optional[bool] = None,
@@ -151,7 +150,6 @@ def execute_dag(
         self,
         dag: MultiDiGraph,
         callbacks: Optional[Sequence[Callback]] = None,
-        array_names: Optional[Sequence[str]] = None,
         resume: Optional[bool] = None,
         spec: Optional[Spec] = None,
         compute_kwargs: Optional[Dict[str, Any]] = None,
@@ -162,7 +160,6 @@ def execute_dag(
             async_execute_dag(
                 dag,
                 callbacks=callbacks,
-                array_names=array_names,
                 resume=resume,
                 spec=spec,
                 compute_kwargs=compute_kwargs,

diff --git a/cubed/runtime/executors/lithops.py b/cubed/runtime/executors/lithops.py
@@ -160,7 +160,6 @@ def map_unordered(
 def execute_dag(
     dag: MultiDiGraph,
     callbacks: Optional[Sequence[Callback]] = None,
-    array_names: Optional[Sequence[str]] = None,
     resume: Optional[bool] = None,
     spec: Optional[Spec] = None,
     compute_arrays_in_parallel: Optional[bool] = None,
@@ -241,7 +240,6 @@ def execute_dag(
         self,
         dag: MultiDiGraph,
         callbacks: Optional[Sequence[Callback]] = None,
-        array_names: Optional[Sequence[str]] = None,
         resume: Optional[bool] = None,
         spec: Optional[Spec] = None,
         **kwargs,
@@ -250,7 +248,6 @@ def execute_dag(
         execute_dag(
             dag,
             callbacks=callbacks,
-            array_names=array_names,
             resume=resume,
             spec=spec,
             **merged_kwargs,

diff --git a/cubed/runtime/executors/modal.py b/cubed/runtime/executors/modal.py
@@ -112,7 +112,6 @@ def run_remotely(self, input, func=None, config=None):
 def execute_dag(
     dag: MultiDiGraph,
     callbacks: Optional[Sequence[Callback]] = None,
-    array_names: Optional[Sequence[str]] = None,
     resume: Optional[bool] = None,
     spec: Optional[Spec] = None,
     cloud: Optional[str] = None,
@@ -151,7 +150,6 @@ def execute_dag(
         self,
         dag: MultiDiGraph,
         callbacks: Optional[Sequence[Callback]] = None,
-        array_names: Optional[Sequence[str]] = None,
         resume: Optional[bool] = None,
         spec: Optional[Spec] = None,
         **kwargs,
@@ -160,7 +158,6 @@ def execute_dag(
         execute_dag(
             dag,
             callbacks=callbacks,
-            array_names=array_names,
             resume=resume,
             spec=spec,
             **merged_kwargs,

diff --git a/cubed/runtime/executors/modal_async.py b/cubed/runtime/executors/modal_async.py
@@ -108,7 +108,6 @@ def pipeline_to_stream(app_function, name, pipeline, **kwargs):
 async def async_execute_dag(
     dag: MultiDiGraph,
     callbacks: Optional[Sequence[Callback]] = None,
-    array_names: Optional[Sequence[str]] = None,
     resume: Optional[bool] = None,
     spec: Optional[Spec] = None,
     cloud: Optional[str] = None,
@@ -155,7 +154,6 @@ def execute_dag(
         self,
         dag: MultiDiGraph,
         callbacks: Optional[Sequence[Callback]] = None,
-        array_names: Optional[Sequence[str]] = None,
         resume: Optional[bool] = None,
         spec: Optional[Spec] = None,
         **kwargs,
@@ -165,7 +163,6 @@ def execute_dag(
             async_execute_dag(
                 dag,
                 callbacks=callbacks,
-                array_names=array_names,
                 resume=resume,
                 spec=spec,
                 **merged_kwargs,

diff --git a/cubed/runtime/executors/python.py b/cubed/runtime/executors/python.py
@@ -18,7 +18,6 @@ def execute_dag(
         self,
         dag: MultiDiGraph,
         callbacks: Optional[Sequence[Callback]] = None,
-        array_names: Optional[Sequence[str]] = None,
         resume: Optional[bool] = None,
         spec: Optional[Spec] = None,
         **kwargs,

diff --git a/cubed/runtime/executors/python_async.py b/cubed/runtime/executors/python_async.py
@@ -82,7 +82,6 @@ def pipeline_to_stream(
 async def async_execute_dag(
     dag: MultiDiGraph,
     callbacks: Optional[Sequence[Callback]] = None,
-    array_names: Optional[Sequence[str]] = None,
     resume: Optional[bool] = None,
     spec: Optional[Spec] = None,
     compute_arrays_in_parallel: Optional[bool] = None,
@@ -125,7 +124,6 @@ def execute_dag(
         self,
         dag: MultiDiGraph,
         callbacks: Optional[Sequence[Callback]] = None,
-        array_names: Optional[Sequence[str]] = None,
         resume: Optional[bool] = None,
         spec: Optional[Spec] = None,
         **kwargs,
@@ -134,7 +132,6 @@ def execute_dag(
             async_execute_dag(
                 dag,
                 callbacks=callbacks,
-                array_names=array_names,
                 resume=resume,
                 spec=spec,
                 **kwargs,