[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

tests/model_explainability/nemo_guardrails/conftest.py

@@ -13,15 +13,14 @@
 from ocp_resources.secret import Secret
 
 from tests.model_explainability.nemo_guardrails.constants import PresidioEntity
-from utilities.constants import LLMdInferenceSimConfig
 from tests.model_explainability.nemo_guardrails.utils import (
     create_llm_judge_config,
     create_presidio_config,
 )
 from utilities.certificates_utils import create_ca_bundle_file
+from utilities.constants import LLMdInferenceSimConfig
 from utilities.nemo_guardrails_resource import NemoGuardrails
 
-
 # ===========================
 # Secret Fixtures
 # ===========================
@@ -242,15 +241,12 @@ def nemo_guardrails_presidio(
             }
         ],
         replicas=1,
-        env=[{
-            "name": "OPENAI_API_KEY",
-            "valueFrom": {
-                "secretKeyRef": {
-                    "name": nemo_api_token_secret.name,
-                    "key": "token"
-                }
-            },
-        }],
+        env=[
+            {
+                "name": "OPENAI_API_KEY",
+                "valueFrom": {"secretKeyRef": {"name": nemo_api_token_secret.name, "key": "token"}},
+            }
+        ],
     ) as nemo_cr:
         # Wait for the deployment to be ready
         deployment = Deployment(
@@ -289,10 +285,12 @@ def nemo_guardrails_multi_config(
             },
         ],
         replicas=1,
-        env=[{
-            "name": "OPENAI_API_KEY",
-            "valueFrom": {"secretKeyRef": {"name": nemo_api_token_secret.name, "key": "token"}},
-        }],
+        env=[
+            {
+                "name": "OPENAI_API_KEY",
+                "valueFrom": {"secretKeyRef": {"name": nemo_api_token_secret.name, "key": "token"}},
+            }
+        ],
     ) as nemo_cr:
         # Wait for the deployment to be ready
         deployment = Deployment(
@@ -325,10 +323,12 @@ def nemo_guardrails_second_server(
             }
         ],
         replicas=1,
-        env=[{
-            "name": "OPENAI_API_KEY",
-            "valueFrom": {"secretKeyRef": {"name": nemo_api_token_secret.name, "key": "token"}},
-        }],
+        env=[
+            {
+                "name": "OPENAI_API_KEY",
+                "valueFrom": {"secretKeyRef": {"name": nemo_api_token_secret.name, "key": "token"}},
+            }
+        ],
     ) as nemo_cr:
         # Wait for the deployment to be ready
         deployment = Deployment(
@@ -435,7 +435,6 @@ def nemo_guardrails_llm_judge_healthcheck(
     Note: The deployment readiness is already verified in the nemo_guardrails_llm_judge fixture
     via deployment.wait_for_replicas(). This fixture just ensures dependencies are created.
     """
-    pass
 
 
 @pytest.fixture(scope="class")
@@ -449,7 +448,6 @@ def nemo_guardrails_presidio_healthcheck(
     Note: The deployment readiness is already verified in the nemo_guardrails_presidio fixture
     via deployment.wait_for_replicas(). This fixture just ensures dependencies are created.
     """
-    pass
 
 
 @pytest.fixture(scope="class")
@@ -463,7 +461,6 @@ def nemo_guardrails_multi_config_healthcheck(
     Note: The deployment readiness is already verified in the nemo_guardrails_multi_config fixture
     via deployment.wait_for_replicas(). This fixture just ensures dependencies are created.
     """
-    pass
 
 
 @pytest.fixture(scope="class")
@@ -477,4 +474,3 @@ def nemo_guardrails_second_server_healthcheck(
     Note: The deployment readiness is already verified in the nemo_guardrails_second_server fixture
     via deployment.wait_for_replicas(). This fixture just ensures dependencies are created.
     """
-    pass

tests/model_explainability/nemo_guardrails/test_nemo_guardrails.py

@@ -11,8 +11,8 @@
     CHAT_ENDPOINT,
     CHECK_ENDPOINT,
     CLEAN_PROMPT,
-    PII_PROMPT,
     MODEL_NAME,
+    PII_PROMPT,
     SAFE_PROMPTS,
 )
 from tests.model_explainability.nemo_guardrails.utils import (
@@ -345,7 +345,7 @@ def test_nemo_multi_config(
             ca_bundle_file=openshift_ca_bundle_file,
             message=SAFE_PROMPTS[0],
             model=MODEL_NAME,
-            configuration="config-a"
+            configuration="config-a",
         )
 
         assert response.status_code == 200, f"Expected 200, got {response.status_code}"
@@ -355,7 +355,7 @@ def test_nemo_multi_config(
         if endpoint == CHAT_ENDPOINT:
             assert "choices" in response_json, "Chat endpoint should contain choices"
             assert "guardrails" in response_json, "Response should contain guardrails metadata"
-            assert response_json["guardrails"]['config_id'] == "config-a", "Should use config-a"
+            assert response_json["guardrails"]["config_id"] == "config-a", "Should use config-a"
         elif endpoint == CHECK_ENDPOINT:
             assert "status" in response_json, "Check endpoint should contain status"
             # For checks endpoint, config selection is validated through the request params
@@ -367,7 +367,7 @@ def test_nemo_multi_config(
             ca_bundle_file=openshift_ca_bundle_file,
             message=PII_PROMPT,
             model=MODEL_NAME,
-            configuration="config-b"
+            configuration="config-b",
         )
 
         assert response.status_code == 200, f"Expected 200, got {response.status_code}"
@@ -377,12 +377,11 @@ def test_nemo_multi_config(
         if endpoint == CHAT_ENDPOINT:
             assert "choices" in response_json, "Chat endpoint should contain choices"
             assert "guardrails" in response_json, "Response should contain guardrails metadata"
-            assert response_json["guardrails"]['config_id'] == "config-b", "Should use config-b"
+            assert response_json["guardrails"]["config_id"] == "config-b", "Should use config-b"
         elif endpoint == CHECK_ENDPOINT:
             assert "status" in response_json, "Check endpoint should contain status"
             # For checks endpoint, config selection is validated through the request params
 
-
     def test_nemo_multi_config_mount_paths(
         self,
         admin_client: DynamicClient,

tests/model_explainability/nemo_guardrails/utils.py

@@ -198,6 +198,7 @@ def create_chat_completion_request(message: str, model: str, configuration: str
 
     return payload
 
+
 def send_request(
     url: str,
     token: str | None,