From 9a19338ce9adc14a19a36ae078c6d1c6a0b7496d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Luk=C3=A1=C5=A1=20Pleva=C4=8D?= <lukas@plevac.eu>
Date: Tue, 29 Oct 2024 07:57:13 +0100
Subject: [PATCH] Aktualizovat versions/slave/airootfs/discover.py

---
 versions/slave/airootfs/discover.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/versions/slave/airootfs/discover.py b/versions/slave/airootfs/discover.py
index b8f7478..64ee89e 100644
--- a/versions/slave/airootfs/discover.py
+++ b/versions/slave/airootfs/discover.py
@@ -52,6 +52,8 @@ def selfInfo():
 
 
 def loadHosts():
+    global HOSTS, MASTER
+
     try:
         with open('/etc/slurm-llnl/hosts.json', 'r') as file:
             HOSTS = json.load(file)
@@ -70,6 +72,9 @@ def updateHosts():
         json.dump(HOSTS, outfile)
 
 def generateSlurmConfig(source, target):
+    if MASTER is None:
+        return
+
     hosts = f"NodeName={socket.gethostname()} NodeAddr={get_ip()} CPUs={os.cpu_count()} State=UNKNOWN\n" # first is my self
     noMasterHosts = ""
     for host in HOSTS.values():