Update granulate_utils/metadata/cloud.py

Co-authored-by: Yonatan Goldschmidt <[email protected]>
Granulate · Jul 10, 2023 · 88e32d3 · 88e32d3
1 parent 3ee53f9
commit 88e32d3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/granulate_utils/metadata/cloud.py b/granulate_utils/metadata/cloud.py
@@ -103,7 +103,9 @@ def get_aws_instance_metadata() -> Optional[AwsInstanceMetadata]:
 
 
 def get_aws_container_metadata() -> Optional[AwsContainerMetadata]:
-    ecs_container_metadata_uri_v4 = os.environ["ECS_CONTAINER_METADATA_URI_V4"]
+    ecs_container_metadata_uri_v4 = os.environ.get("ECS_CONTAINER_METADATA_URI_V4")
+    if ecs_container_metadata_uri_v4 is None:
+        return None
     response = send_request(ecs_container_metadata_uri_v4)
     if response is None:
         return None