Fixed is_main_thread

SLURM_PROCID is a string, not an integer
metno · Jan 15, 2025 · 9e91ccf · 9e91ccf
1 parent fb28d64
commit 9e91ccf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bris/__main__.py b/bris/__main__.py
@@ -80,7 +80,7 @@ def main():
     inference.run()
 
     # Finalize all output, so they can flush to disk if needed
-    is_main_thread = ("SLURM_PROCID" not in os.environ) or (os.environ["SLURM_PROCID"] == 0)
+    is_main_thread = ("SLURM_PROCID" not in os.environ) or (os.environ["SLURM_PROCID"] == "0")
     if is_main_thread:
         for decoder_output in decoder_outputs:
             for output in decoder_output["outputs"]: