fix: tests

DataChefHQ · Oct 10, 2024 · 7b0863f · 7b0863f
1 parent 6fe1bcf
commit 7b0863f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/damavand/cloud/aws/controllers/llm.py b/src/damavand/cloud/aws/controllers/llm.py
@@ -108,7 +108,7 @@ def resource(self) -> PulumiResource:
             name=self.name,
             args=AwsVllmComponentArgs(
                 region=self._region,
-                public_internet_access=True,
+                api_key_required=True,
                 endpoint_ssm_parameter_name=self._base_url_ssm_name,
             ),
         )
diff --git a/tests/clouds/aws/resources/test_vllm_component.py b/tests/clouds/aws/resources/test_vllm_component.py
@@ -27,7 +27,7 @@ def call(self, args: MockCallArgs) -> Tuple[dict, Optional[List[Tuple[str, str]]
 )
 
 
-def test_private_internet_access():
+def test_require_api_key():
     vllm = AwsVllmComponent(
         name="test",
         args=AwsVllmComponentArgs(),