Skip to content

Commit 4458c86

Browse files
authored
create_file_executor() - rename backend variable (#713)
* create_file_executor() - rename backend variable * change default * Add option to set execute_function
1 parent 957170b commit 4458c86

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

executorlib/executor/flux.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -348,7 +348,7 @@ def __init__(
348348
super().__init__(
349349
executor=create_file_executor(
350350
max_workers=max_workers,
351-
backend="flux_submission",
351+
backend="flux",
352352
max_cores=max_cores,
353353
cache_directory=cache_directory,
354354
resource_dict=resource_dict,

executorlib/executor/slurm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def __init__(
160160
super().__init__(
161161
executor=create_file_executor(
162162
max_workers=max_workers,
163-
backend="slurm_submission",
163+
backend="slurm",
164164
max_cores=max_cores,
165165
cache_directory=cache_directory,
166166
resource_dict=resource_dict,

executorlib/task_scheduler/file/task_scheduler.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def __init__(
8080
def create_file_executor(
8181
resource_dict: dict,
8282
max_workers: Optional[int] = None,
83-
backend: str = "flux_submission",
83+
backend: Optional[str] = None,
8484
max_cores: Optional[int] = None,
8585
cache_directory: Optional[str] = None,
8686
flux_executor=None,
@@ -92,6 +92,7 @@ def create_file_executor(
9292
block_allocation: bool = False,
9393
init_function: Optional[Callable] = None,
9494
disable_dependencies: bool = False,
95+
execute_function: Callable = execute_with_pysqa,
9596
):
9697
if block_allocation:
9798
raise ValueError(
@@ -112,6 +113,7 @@ def create_file_executor(
112113
return FileTaskScheduler(
113114
resource_dict=resource_dict,
114115
pysqa_config_directory=pysqa_config_directory,
115-
backend=backend.split("_submission")[0],
116+
backend=backend,
116117
disable_dependencies=disable_dependencies,
118+
execute_function=execute_function,
117119
)

0 commit comments

Comments
 (0)