Update find_machine.py for CEE (sandialabs#563)

* Update find_machine.py for CEE * Update find_machine.py
psakievich · Sep 6, 2023 · c995b17 · c995b17
1 parent 258a590
commit c995b17
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/spack-scripting/scripting/cmd/manager_cmds/find_machine.py b/spack-scripting/scripting/cmd/manager_cmds/find_machine.py
@@ -17,9 +17,14 @@ def __init__(self, test, full_machine_name):
 
 
 def is_cee(hostname):
+    site = False
+    system = False
     if "SNLSITE" in os.environ:
-        return os.environ["SNLSITE"] == "cee"
-    return False
+        site = os.environ["SNLSITE"] == "cee"
+    if "SNLSYSTEM" in os.environ:
+        system = os.environ["SNLSYSTEM"] == "cee"
+    detected = site or system
+    return detected
 
 
 def is_snl_hpc(hostname):