Update deephyper node list util (ORNL#261)

RylieWeaver · Sep 25, 2024 · 5c8f5e5 · 5c8f5e5
1 parent e3475fd
commit 5c8f5e5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/hydragnn/utils/deephyper.py b/hydragnn/utils/deephyper.py
@@ -12,6 +12,10 @@ def master_from_host(host):
 
 def read_node_list():
     node_list = os.environ["SLURM_NODELIST"]
+    if not "[" in node_list:
+        return [
+            node_list,
+        ], node_list
     nodes = []
     system = os.getenv("HYDRAGNN_SYSTEM", "frontier")
     if system == "frontier":