[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_registry/model_catalog/test_model_artifact_search.py

@@ -17,6 +17,7 @@
 pytestmark = [pytest.mark.usefixtures("updated_dsc_component_state_scope_session", "model_registry_namespace")]
 MODEL_NAME_ARTIFACT_SEARCH = "RedHatAI/Llama-3.1-8B-Instruct"
 
+
 class TestSearchArtifactsByFilterQuery:
     @pytest.mark.parametrize(
         "randomly_picked_model_from_catalog_api_by_source, invalid_filter_query",
@@ -45,7 +46,7 @@ def test_search_artifacts_by_invalid_filter_query(
         model_catalog_rest_url: list[str],
         model_registry_rest_headers: dict[str, str],
         randomly_picked_model_from_catalog_api_by_source: tuple[dict, str, str],
-            invalid_filter_query: str,
+        invalid_filter_query: str,
     ):
         """
         Tests the API's response to invalid filter queries syntax when searching artifacts.
@@ -64,38 +65,52 @@ def test_search_artifacts_by_invalid_filter_query(
                 page_size=1,
             )
 
-        LOGGER.info(f"Successfully validated that invalid artifact filter query '{invalid_filter_query}' raises an error")
+        LOGGER.info(
+            f"Successfully validated that invalid artifact filter query '{invalid_filter_query}' raises an error"
+        )
 
     @pytest.mark.parametrize(
         "randomly_picked_model_from_catalog_api_by_source, filter_query, expected_value, logic_type",
         [
             pytest.param(
-                {"catalog_id": VALIDATED_CATALOG_ID, "header_type": "registry",
-                 "model_name": MODEL_NAME_ARTIFACT_SEARCH},
+                {
+                    "catalog_id": VALIDATED_CATALOG_ID,
+                    "header_type": "registry",
+                    "model_name": MODEL_NAME_ARTIFACT_SEARCH,
+                },
                 "hardware_type.string_value = 'ABC-1234'",
                 None,
                 None,
                 id="test_valid_artifact_filter_query_no_results",
             ),
             pytest.param(
-                {"catalog_id": VALIDATED_CATALOG_ID, "header_type": "registry",
-                 "model_name": MODEL_NAME_ARTIFACT_SEARCH},
+                {
+                    "catalog_id": VALIDATED_CATALOG_ID,
+                    "header_type": "registry",
+                    "model_name": MODEL_NAME_ARTIFACT_SEARCH,
+                },
                 "requests_per_second.double_value > 15.0",
                 [{"key_name": "requests_per_second", "key_type": "double_value", "comparison": "min", "value": 15.0}],
                 "and",
                 id="test_performance_min_filter",
             ),
             pytest.param(
-                {"catalog_id": VALIDATED_CATALOG_ID, "header_type": "registry",
-                 "model_name": MODEL_NAME_ARTIFACT_SEARCH},
+                {
+                    "catalog_id": VALIDATED_CATALOG_ID,
+                    "header_type": "registry",
+                    "model_name": MODEL_NAME_ARTIFACT_SEARCH,
+                },
                 "hardware_count.int_value = 8",
                 [{"key_name": "hardware_count", "key_type": "int_value", "comparison": "exact", "value": 8}],
                 "and",
                 id="test_hardware_exact_filter",
             ),
             pytest.param(
-                {"catalog_id": VALIDATED_CATALOG_ID, "header_type": "registry",
-                 "model_name": MODEL_NAME_ARTIFACT_SEARCH},
+                {
+                    "catalog_id": VALIDATED_CATALOG_ID,
+                    "header_type": "registry",
+                    "model_name": MODEL_NAME_ARTIFACT_SEARCH,
+                },
                 "(hardware_type.string_value = 'H100') AND (ttft_p99.double_value < 200)",
                 [
                     {"key_name": "hardware_type", "key_type": "string_value", "comparison": "exact", "value": "H100"},
@@ -105,8 +120,11 @@ def test_search_artifacts_by_invalid_filter_query(
                 id="test_combined_hardware_performance_filter_and_operation",
             ),
             pytest.param(
-                {"catalog_id": VALIDATED_CATALOG_ID, "header_type": "registry",
-                 "model_name": MODEL_NAME_ARTIFACT_SEARCH},
+                {
+                    "catalog_id": VALIDATED_CATALOG_ID,
+                    "header_type": "registry",
+                    "model_name": MODEL_NAME_ARTIFACT_SEARCH,
+                },
                 "(tps_mean.double_value <260) OR (hardware_type.string_value = 'A100-80')",
                 [
                     {"key_name": "tps_mean", "key_type": "double_value", "comparison": "max", "value": 260},
@@ -150,10 +168,12 @@ def test_filter_query_advanced_artifact_search(
         )
 
         if expected_value is None:
-        # Simple validation of length and size for basic filter queries
+            # Simple validation of length and size for basic filter queries
             assert len(result["items"]) >= 0, f"Filter query '{filter_query}' should return valid results"
             assert result["size"] >= 0, f"Size should be non-negative for filter query '{filter_query}'"
-            LOGGER.info(f"Successfully validated that filter query '{filter_query}' returns {len(result['items'])} artifacts")
+            LOGGER.info(
+                f"Successfully validated that filter query '{filter_query}' returns {len(result['items'])} artifacts"
+            )
         else:
             # Advanced validation using criteria matching
             all_artifacts = result["items"]
@@ -172,6 +192,8 @@ def test_filter_query_advanced_artifact_search(
                 raise ValueError(f"Invalid logic_type: {logic_type}. Must be 'and' or 'or'")
 
             if validation_result:
-                LOGGER.info(f"For Model: {model_name}, {logic_type.upper()} validation completed successfully for {len(all_artifacts)} artifacts")
+                LOGGER.info(
+                    f"For Model: {model_name}, {logic_type.upper()} validation completed successfully for {len(all_artifacts)} artifacts"
+                )
             else:
-                assert False, f"{logic_type.upper()} filter validation failed for model {model_name}"
+                assert False, f"{logic_type.upper()} filter validation failed for model {model_name}"