add support for custom exec_mode

cadCAD-org · Mar 9, 2023 · d8d7e2c · d8d7e2c
1 parent 14fe8b1
commit d8d7e2c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/cadCAD_tools/execution/easy_run.py b/cadCAD_tools/execution/easy_run.py
@@ -21,7 +21,8 @@ def easy_run(state_variables,
              N_samples,
              use_label=False,
              assign_params: Union[bool, set]=True,
-             drop_substeps=True) -> pd.DataFrame:
+             drop_substeps=True,
+             exec_mode='local') -> pd.DataFrame:
     """
     Run cadCAD simulations without headaches.
     """
@@ -42,8 +43,11 @@ def easy_run(state_variables,
     configs = exp.configs
 
     # Set-up cadCAD executor
-    exec_mode = ExecutionMode()
-    exec_context = ExecutionContext(exec_mode.local_mode)
+    if exec_mode == 'local':
+        _exec_mode = ExecutionMode().local_mode
+    elif exec_mode == 'single':
+        _exec_mode = ExecutionMode().single_mode
+    exec_context = ExecutionContext(_exec_mode)
     executor = Executor(exec_context=exec_context, configs=configs)
 
     # Execute the cadCAD experiment

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 """
 
 setup(name='cadCAD_tools',
-      version='0.0.1.6',
+      version='0.0.1.7',
       description="tools",
       long_description=long_description,
       url='https://github.com/cadCAD-org/cadCAD-tools',