From 1795addc2d88f46cb0192d67bdaffb607f55c541 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jord=C3=A0=20Polo?= <jorda.polo@amd.com>
Date: Mon, 8 Jul 2024 14:38:21 -0700
Subject: [PATCH] Fix slurm environment script

---
 omnistat/slurm_env.py             | 6 +++---
 test/docker/slurm/slurm-prolog.sh | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/omnistat/slurm_env.py b/omnistat/slurm_env.py
index 0e861809..56bda8e2 100755
--- 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
index e759a376..6112c3a7 100755
--- 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