bugfix in launcher logic

molmod · Feb 10, 2024 · 208252a · 208252a
1 parent 0975d0c
commit 208252a
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/psiflow/execution.py b/psiflow/execution.py
@@ -246,10 +246,7 @@ def parse_config(yaml_dict: dict):
                 provider_dict = _dict.pop(provider_keys[0])
 
                 # if provider requests multiple nodes, switch to (containerized) SrunLauncher
-                if (
-                    provider_dict.pop("nodes_per_block", 1) > 1
-                    and "container" in yaml_dict
-                ):
+                if provider_dict.pop("nodes_per_block", 1) > 1:
                     assert (
                         provider_keys[0] == "SlurmProvider"
                     ), "multi-node blocks only supported for SLURM"