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