bugfix

molmod · Feb 10, 2024 · faba588 · faba588
1 parent 208252a
commit faba588
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/psiflow/execution.py b/psiflow/execution.py
@@ -246,7 +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:
+                if provider_dict.get("nodes_per_block", 1) > 1:
                     assert (
                         provider_keys[0] == "SlurmProvider"
                     ), "multi-node blocks only supported for SLURM"