From ffd6b376fbbc7fd189801e9fb4fc9afff5140fed Mon Sep 17 00:00:00 2001 From: Jathavaan Shankarr Date: Tue, 26 May 2026 15:26:41 +0200 Subject: [PATCH] #355 Fix missing AZURE_UAMI_RESOURCE_ID in ACI containers and quota leak Forward AZURE_UAMI_RESOURCE_ID to container environment variables so Config initialization no longer crashes on startup. Wrap container state polling in try/finally so failed containers are always cleaned up, preventing ACI core quota exhaustion across batches. --- main.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 1eaf59a5..7324d0d1 100644 --- a/main.py +++ b/main.py @@ -193,8 +193,10 @@ def _run_container_benchmark( memory_gb=memory_gb, dataset_size=dataset_size, ) - _check_container_state(container_group_name=container_group_name) - _delete_container_instance(container_group_name=container_group_name) + try: + _check_container_state(container_group_name=container_group_name) + finally: + _delete_container_instance(container_group_name=container_group_name) def _create_run_id() -> str: @@ -377,6 +379,7 @@ def _create_container_instance( Config.AZURE_UAMI_RESOURCE_ID, "--environment-variables", f"AZURE_SUBSCRIPTION_ID={Config.AZURE_SUBSCRIPTION_ID}", + f"AZURE_UAMI_RESOURCE_ID={Config.AZURE_UAMI_RESOURCE_ID}", f"AZURE_BLOB_STORAGE_BENCHMARK_CONTAINER={StorageContainer.BENCHMARKS.value}", f"AZURE_BLOB_STORAGE_METADATA_CONTAINER={StorageContainer.METADATA.value}", f"POSTGRES_SERVER_NAME={Config.POSTGRES_SERVER_NAME}",