set env vars multiple times

dfm · Jul 28, 2023 · 121b4ce · 121b4ce
1 parent d1a9d6a
commit 121b4ce
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/tess_atlas/__init__.py b/src/tess_atlas/__init__.py
@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+from .utils import set_global_environ_vars
 
 __all__ = []
 
@@ -10,3 +11,5 @@
 __copyright__ = "Copyright 2020 TESS Atlas developers"
 __contributors__ = "https://github.com/dfm/tess-atlas/graphs/contributors"
 __website__ = "http://catalog.tess-atlas.cloud.edu.au"
+
+set_global_environ_vars()
diff --git a/src/tess_atlas/slurm_job_generator/templates/slurm_template.sh b/src/tess_atlas/slurm_job_generator/templates/slurm_template.sh
@@ -21,4 +21,6 @@ module load {{module_loads}}
 ARRAY_ARGS=({{array_args}})
 {% endif %}
 echo "Job tmp path: $JOBFS"
+export THEANO_FLAGS="base_compiledir=$JOBFS/.theano_base,compiledir=$JOBFS/.theano_compile"
+export IPYTHONDIR=$JOBFS/.ipython
 {{command}}