Update rp_scale.py

runpod · Nov 20, 2023 · 88271e0 · 88271e0
1 parent 25ed917
commit 88271e0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/runpod/serverless/modules/rp_scale.py b/runpod/serverless/modules/rp_scale.py
@@ -82,8 +82,11 @@ class JobScaler():
     """
 
     # , concurrency_controller: typing.Any, config: Dict[str, typing.Any]):
-    def __init__(self, concurrency_modifier: typing.Any = _default_concurrency_modifier):
-        self.concurrency_modifier = concurrency_modifier
+    def __init__(self, concurrency_modifier: typing.Any):
+        if concurrency_modifier is None:
+            concurrency_modifier = _default_concurrency_modifier
+        else:
+            self.concurrency_modifier = concurrency_modifier
         # self.concurrency_controller = concurrency_controller
 
         # self.config = ConcurrencyConfig(