From 547077317a704319a9b4604952fd60c1dd5a29b0 Mon Sep 17 00:00:00 2001 From: Dodam Ih Date: Wed, 8 Nov 2023 13:58:55 -0800 Subject: [PATCH] hotfix: remove unused max_pull_num, missing import --- zetta_utils/cloud_management/resource_allocation/k8s/common.py | 1 - zetta_utils/mazepa_addons/configurations/__init__.py | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/zetta_utils/cloud_management/resource_allocation/k8s/common.py b/zetta_utils/cloud_management/resource_allocation/k8s/common.py index f405b65a4..015e06c1e 100644 --- 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 index c9297e123..dcd38ce51 100644 --- 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