Trying a different tmp/scratch folder

terrapower · Feb 27, 2025 · fa80df4 · fa80df4
1 parent 228c2b0
commit fa80df4
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/armi/context.py b/armi/context.py
@@ -109,14 +109,19 @@ def setMode(cls, mode):
     # stick with defaults
     pass
 
-try:
+if sys.platform.startswith("win"):
     # trying a Windows approach
     APP_DATA = os.path.join(os.environ["APPDATA"], "armi")
     APP_DATA = APP_DATA.replace("/", "\\")
-except Exception:
+else:
     # non-Windows
     APP_DATA = os.path.expanduser("~/.armi")
 
+    if os.access("/mnt/scratch/", os.W_OK):
+        APP_DATA = "/mnt/scratch/.armi"
+    elif os.access("/tmp/", os.W_OK):
+        APP_DATA = "/tmp/.armi"
+
 if MPI_NODENAMES.index(MPI_NODENAME) == MPI_RANK:
     if not os.path.isdir(APP_DATA):
         try: