diff --git a/executorlib/executor/flux.py b/executorlib/executor/flux.py index a41f0f47..145ccc73 100644 --- a/executorlib/executor/flux.py +++ b/executorlib/executor/flux.py @@ -285,7 +285,6 @@ def __init__( plot_dependency_graph: bool = False, plot_dependency_graph_filename: Optional[str] = None, log_obj_size: bool = False, - terminate_tasks_on_shutdown: bool = True, ): """ The executorlib.FluxClusterExecutor leverages either the message passing interface (MPI), the SLURM workload diff --git a/executorlib/executor/slurm.py b/executorlib/executor/slurm.py index 6ceac729..bf3e48c7 100644 --- a/executorlib/executor/slurm.py +++ b/executorlib/executor/slurm.py @@ -97,7 +97,6 @@ def __init__( plot_dependency_graph: bool = False, plot_dependency_graph_filename: Optional[str] = None, log_obj_size: bool = False, - terminate_tasks_on_shutdown: bool = True, ): """ The executorlib.SlurmClusterExecutor leverages either the message passing interface (MPI), the SLURM workload @@ -281,7 +280,6 @@ def __init__( plot_dependency_graph: bool = False, plot_dependency_graph_filename: Optional[str] = None, log_obj_size: bool = False, - terminate_tasks_on_shutdown: bool = True, ): """ The executorlib.SlurmJobExecutor leverages either the message passing interface (MPI), the SLURM workload diff --git a/tests/test_fluxclusterexecutor.py b/tests/test_fluxclusterexecutor.py index 3e27d5e1..cfb838aa 100644 --- a/tests/test_fluxclusterexecutor.py +++ b/tests/test_fluxclusterexecutor.py @@ -37,7 +37,6 @@ def test_executor(self): resource_dict={"cores": 2, "cwd": "executorlib_cache"}, block_allocation=False, cache_directory="executorlib_cache", - terminate_tasks_on_shutdown=False, ) as exe: cloudpickle_register(ind=1) fs1 = exe.submit(mpi_funct, 1) @@ -51,7 +50,6 @@ def test_executor_no_cwd(self): resource_dict={"cores": 2}, block_allocation=False, cache_directory="executorlib_cache", - terminate_tasks_on_shutdown=True, ) as exe: cloudpickle_register(ind=1) fs1 = exe.submit(mpi_funct, 1) diff --git a/tests/test_slurmclusterexecutor.py b/tests/test_slurmclusterexecutor.py index aa4f2c67..41d0f94b 100644 --- a/tests/test_slurmclusterexecutor.py +++ b/tests/test_slurmclusterexecutor.py @@ -50,7 +50,6 @@ def test_executor(self): resource_dict={"cores": 2, "cwd": "executorlib_cache", "submission_template": submission_template}, block_allocation=False, cache_directory="executorlib_cache", - terminate_tasks_on_shutdown=False, ) as exe: cloudpickle_register(ind=1) fs1 = exe.submit(mpi_funct, 1) @@ -64,7 +63,6 @@ def test_executor_no_cwd(self): resource_dict={"cores": 2, "submission_template": submission_template}, block_allocation=False, cache_directory="executorlib_cache", - terminate_tasks_on_shutdown=True, ) as exe: cloudpickle_register(ind=1) fs1 = exe.submit(mpi_funct, 1)