Fix slurm environment script

AMDResearch · Jul 8, 2024 · d37edd8 · d37edd8
1 parent d7f7827
commit d37edd8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/omnistat/slurm_env.py b/omnistat/slurm_env.py
@@ -35,11 +35,11 @@
 import os
 import sys
 
-jobData = {}
-jobFile = "/tmp/omni_slurmjobinfo"
-
 
 def main():
+    jobData = {}
+    jobFile = "/tmp/omni_slurmjobinfo"
+
     if len(sys.argv) > 1:
         jobFile = sys.argv[1]
 

diff --git a/test/docker/slurm/slurm-prolog.sh b/test/docker/slurm/slurm-prolog.sh
@@ -1,3 +1,3 @@
 #!/bin/bash
 
-/opt/omnistat/bin/python -m omnistat.slurm_env
+/opt/omnistat/bin/omnistat-slurm-env