fix: tests

DataChefHQ · Oct 9, 2024 · 6fe1bcf · 6fe1bcf
1 parent 0fbe6d8
commit 6fe1bcf
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/src/damavand/cloud/aws/resources/vllm_component.py b/src/damavand/cloud/aws/resources/vllm_component.py
@@ -423,7 +423,7 @@ def api_key_secret_version(self) -> aws.secretsmanager.SecretVersion:
         AttributeError
             When api_key_required is False.
         """
-        if self.args.api_key_required:
+        if not self.args.api_key_required:
             raise AttributeError("`api_key_secret_version` is only available when api_key_required is False")
 
         return aws.secretsmanager.SecretVersion(
@@ -445,7 +445,7 @@ def default_usage_plan(self) -> aws.apigateway.UsagePlan:
         AttributeError
             When api_key_required is False.
         """
-        if self.args.api_key_required:
+        if not self.args.api_key_required:
             raise AttributeError("`default_usage_plan` is only available when api_key_required is False")
 
         return aws.apigateway.UsagePlan(
@@ -472,7 +472,7 @@ def tier_1_usage_plan(self) -> aws.apigateway.UsagePlan:
         AttributeError
             When api_key_required is False.
         """
-        if self.args.api_key_required:
+        if not self.args.api_key_required:
             raise AttributeError("`default_usage_plan` is only available when api_key_required is False")
 
         return aws.apigateway.UsagePlan(
@@ -501,7 +501,7 @@ def tier_2_usage_plan(self) -> aws.apigateway.UsagePlan:
         AttributeError
             When api_key_required is False.
         """
-        if self.args.api_key_required:
+        if not self.args.api_key_required:
             raise AttributeError("`default_usage_plan` is only available when api_key_required is False")
 
         return aws.apigateway.UsagePlan(
@@ -530,7 +530,7 @@ def tier_3_usage_plan(self) -> aws.apigateway.UsagePlan:
         AttributeError
             When api_key_required is False.
         """
-        if self.args.api_key_required:
+        if not self.args.api_key_required:
             raise AttributeError("`default_usage_plan` is only available when api_key_required is False")
 
         return aws.apigateway.UsagePlan(
@@ -558,7 +558,7 @@ def api_key_usage_plan(self) -> aws.apigateway.UsagePlanKey:
         AttributeError
             When api_key_required is False.
         """
-        if self.args.api_key_required:
+        if not self.args.api_key_required:
             raise AttributeError("`api_key_usage_plan` is only available when api_key_required is False")
 
         return aws.apigateway.UsagePlanKey(

diff --git a/tests/clouds/aws/resources/test_vllm_component.py b/tests/clouds/aws/resources/test_vllm_component.py
@@ -52,7 +52,7 @@ def test_public_internet_access():
     vllm = AwsVllmComponent(
         name="test",
         args=AwsVllmComponentArgs(
-            public_internet_access=True,
+            api_key_required=False,
         ),
     )
 
@@ -78,7 +78,7 @@ def test_model_image_version():
         name="test",
         args=AwsVllmComponentArgs(
             model_image_version="0.29.0",
-            public_internet_access=True,
+            api_key_required=True,
         ),
     )
 
@@ -90,7 +90,7 @@ def test_model_image_config():
         name="test",
         args=AwsVllmComponentArgs(
             model_name="microsoft/Phi-3-mini-4k-instruct",
-            public_internet_access=True,
+            api_key_required=True,
         ),
     )