Merge pull request #97 from daniel-thom/local-scratch

Handle local scratch on Kestrel

Author: daniel-thom

jade/__init__.py

@@ -2,6 +2,6 @@
 
 import logging
 
-__version__ = "0.10.0"
+__version__ = "0.10.1"
 
 logging.getLogger(__name__).addHandler(logging.NullHandler())

jade/hpc/slurm_manager.py

@@ -3,7 +3,8 @@
 import logging
 import os
 import re
-from datetime import datetime, timedelta
+import tempfile
+from datetime import datetime
 
 from jade.enums import Status
 from jade.exceptions import ExecutionError  # , InvalidConfiguration
@@ -253,7 +254,10 @@ def get_job_stats(self, job_id):
         return stats
 
     def get_local_scratch(self):
-        return os.environ["LOCAL_SCRATCH"]
+        for key in ("LOCAL_SCRATCH", "TMPDIR"):
+            if key in os.environ:
+                return os.environ[key]
+        return tempfile.gettempdir()
 
     def get_node_id(self):
         return os.environ["SLURM_NODEID"]

jade/version.py

@@ -76,10 +76,8 @@ Homepage = "https://github.com./NREL/jade"
 [tool.hatch.version]
 path = "jade/__init__.py"
 
-[tool.hatch.build.targets.sdist]
-include = [
-    "/jade",
-]
+[tool.hatch.build.targets.wheel]
+packages = ["jade"]
 
 [tool.black]
 line-length = 99