hotfix: remove unused max_pull_num, missing import

ZettaAI · Nov 8, 2023 · 5470773 · 5470773
1 parent 34e73e6
commit 5470773
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/zetta_utils/cloud_management/resource_allocation/k8s/common.py b/zetta_utils/cloud_management/resource_allocation/k8s/common.py
@@ -75,7 +75,6 @@ def get_mazepa_worker_command(
         + f"semaphores_spec: {json.dumps(semaphores_spec)}\n"
         + f"num_procs: {num_procs}\n"
         + """
-        max_pull_num: 1
         sleep_sec: 5
     }'
     """

diff --git a/zetta_utils/mazepa_addons/configurations/__init__.py b/zetta_utils/mazepa_addons/configurations/__init__.py
@@ -1,2 +1,3 @@
 from . import execute_on_gcp_with_sqs
 from . import execute_locally
+from . import worker_pool