Add SUPERDSM_NUM_CPUS env variable

BMCV · Feb 20, 2024 · 3bf095a · 3bf095a
1 parent fa27306
commit 3bf095a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/regressiontests.yml b/.github/workflows/regressiontests.yml
@@ -56,6 +56,7 @@ jobs:
             python -m "superdsm.batch" examples --task-dir "${{ matrix.taskdir }}" --run
         env:
           SUPERDSM_INTERMEDIATE_OUTPUT: false
+          SUPERDSM_NUM_CPUS: 10
 
       - name: Validate results
         id: validation

diff --git a/superdsm/batch.py b/superdsm/batch.py
@@ -258,7 +258,7 @@ def _fmt_path(self, path):
     def _initialize(self):
         for key, val in self.environ.items():
             os.environ[key] = str(val)
-        ray.init(num_cpus=self.data['num_cpus'], log_to_driver=False, logging_level='error')
+        ray.init(num_cpus=os.environ.get('SUPERDSM_NUM_CPUS', 2), log_to_driver=False, logging_level='error')
         _pipeline = create_default_pipeline()
         return _pipeline
 

diff --git a/superdsm/version.py b/superdsm/version.py
@@ -1,5 +1,5 @@
 VERSION_MAJOR = 0
-VERSION_MINOR = 2
+VERSION_MINOR = 3
 VERSION_PATCH = 0
 
 VERSION = '%d.%d%s' % (VERSION_MAJOR, VERSION_MINOR, '.%d' % VERSION_PATCH if VERSION_PATCH > 0 else '')