Cast environment variables to integers

glotzerlab · Dec 5, 2024 · 3592944 · 3592944
1 parent 6caa763
commit 3592944
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/doc/src/guide/concepts/hoomd-example.py b/doc/src/guide/concepts/hoomd-example.py
@@ -17,7 +17,7 @@ def action_implementation(job, communicator):
 
 def action(*jobs):
     """Execute actions on directories in parallel using HOOMD-blue."""
-    processes_per_directory = os.environ['ACTION_PROCESSES_PER_DIRECTORY']
+    processes_per_directory = int(os.environ['ACTION_PROCESSES_PER_DIRECTORY'])
     communicator = hoomd.communicator.Communicator(ranks_per_partition=processes_per_directory)
     action_implementation(jobs[communicator.partition], communicator)
     # ANCHOR_END: action

diff --git a/doc/src/guide/concepts/multiprocessing-example.py b/doc/src/guide/concepts/multiprocessing-example.py
@@ -15,7 +15,7 @@ def action_implementation(job):
 
 def action(*jobs):
     """Process any number of jobs in parallel with the multiprocessing package."""
-    processes = os.environ.get('ACTION_THREADS_PER_PROCESS', multiprocessing.cpu_count())
+    processes = int(os.environ.get('ACTION_THREADS_PER_PROCESS', multiprocessing.cpu_count()))
     if hasattr(os, 'sched_getaffinity'):
         processes = len(os.sched_getaffinity(0))