Set num workers to 0

SeldonIO · Jun 19, 2024 · 752611b · 752611b
1 parent 3733b61
commit 752611b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/runtimes/alibi-explain/tests/conftest.py b/runtimes/alibi-explain/tests/conftest.py
@@ -287,7 +287,7 @@ async def integrated_gradients_runtime(tf_mnist_model_uri: str) -> AlibiExplainR
 
     rt = AlibiExplainRuntime(
         ModelSettings(
-            parallel_workers=1,
+            parallel_workers=0,
             implementation=AlibiExplainRuntime,
             parameters=ModelParameters(extra=explainer_settings.model_dump()),
         )

diff --git a/runtimes/mlflow/tests/conftest.py b/runtimes/mlflow/tests/conftest.py
@@ -86,9 +86,9 @@ def pytorch_model_uri(
         if os.path.exists(model_path):
             return model_path
 
-        model = LightningMNISTClassifier(batch_size=64, num_workers=3, lr=0.001)
+        model = LightningMNISTClassifier(batch_size=64, num_workers=0, lr=0.001)
 
-        dm = MNISTDataModule(batch_size=64, num_workers=3)
+        dm = MNISTDataModule(batch_size=64, num_workers=0)
         dm.setup(stage="fit")
 
         early_stopping = EarlyStopping(

diff --git a/tox.ini b/tox.ini
@@ -37,7 +37,7 @@ set_env =
     GITHUB_REF        = {env:GITHUB_REF:refs/heads/master}
 
 [testenv:{all-runtimes,all-mlserver}]
-env_dir = {toxworkdir}{/}static
+env_dir = {toxworkdir}{/}all-env
 runner = ignore_env_name_mismatch
 commands_pre =
     poetry install --sync --no-root \
@@ -53,7 +53,7 @@ commands =
     all-runtimes:        {toxinidir}/runtimes/lightgbm \
     all-runtimes:        {toxinidir}/runtimes/mlflow \
     all-runtimes:        {toxinidir}/runtimes/huggingface
-    
+
     # kafka and parallel tests are failinig for macos when running in 
     # parallel with the entire test suite. So, we run them separately.
     all-mlserver: python -m pytest {posargs} -n auto \