Merge pull request #2 from chhsiao1981/tmp-dir-for-pytorch

add os.environ['TEMP'] as outputdir/tmp for pytorch tmp-dir

spleenseg/spleenseg.py

@@ -12,7 +12,7 @@
 # from contextlib import redirect_stderr
 # from io import StringIO
 
-# import os
+import os
 import sys
 import re
 import pudb
@@ -98,7 +98,7 @@ def inputFilesSets_trainValidateFind(
     """
     trainingSpace: list[dict[str, str]] = trainingData_prep(options)
     trainingSet: list[dict[str, str]] = trainingSpace[: -options.validateSize]
-    validateSet: list[dict[str, str]] = trainingSpace[-options.validateSize :]
+    validateSet: list[dict[str, str]] = trainingSpace[-options.validateSize:]
     return trainingSet, validateSet
 
 
@@ -222,6 +222,9 @@ def main(options: Namespace, inputdir: Path, outputdir: Path):
     :param outputdir: directory where to write output files
     """
 
+    os.environ['TEMP'] = f'{outputdir}/tmp'
+    os.makedirs(os.environ['TEMP'], exist_ok=True)
+
     # pudb.set_trace()
 
     envDetail_print(options)