Merge pull request #33 from notAI-tech/bedapudi6788-patch-1

Configurable num workers via NUM_WORKERS env variable
preetham · Aug 28, 2020 · c94af6e · c94af6e
2 parents 4da53bf + f62533b
commit c94af6e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -6,7 +6,7 @@ autopep8==1.4.4
 awscli==1.16.196
 babel==2.7.0
 bandit==1.6.2
-bleach==3.1.0
+bleach==3.1.1
 botocore==1.12.186
 certifi==2019.6.16
 chardet==3.0.4

diff --git a/verifytweet/config/settings.py b/verifytweet/config/settings.py
@@ -30,6 +30,9 @@ def no_of_workers():
     twice the cpu count.
 
     """
+    if os.getenv('NUM_WORKERS').isdigit():
+        return int(os.getenv('NUM_WORKERS'))
+
     return multiprocessing.cpu_count() * 2 + 1