Skip to content

Commit 88554ce

Browse files
committed
fix: slurm_native arg handling
1 parent 0ce560d commit 88554ce

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pysr/julia_helpers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,8 @@ def load_cluster_manager(cluster_manager: str) -> AnyValue:
3636
if cluster_manager == "slurm_native":
3737
jl.seval("using SlurmClusterManager: SlurmManager")
3838
# TODO: Is this the right way to do this?
39-
jl.seval("addprocs_slurm_native(; _...) = addprocs(SlurmManager())")
39+
jl.seval(f"using Distributed: addprocs")
40+
jl.seval("addprocs_slurm_native(args...; kws...) = addprocs(SlurmManager())")
4041
return jl.addprocs_slurm_native
4142
elif cluster_manager in KNOWN_CLUSTERMANAGER_BACKENDS:
4243
jl.seval(f"using ClusterManagers: addprocs_{cluster_manager}")

0 commit comments

Comments
 (0)