diff --git a/tests/model_registry/conftest.py b/tests/model_registry/conftest.py index 6323c03e8..79feac33a 100644 --- a/tests/model_registry/conftest.py +++ b/tests/model_registry/conftest.py @@ -185,12 +185,9 @@ def model_registry_instance_service( @pytest.fixture(scope="class") def model_registry_instance_rest_endpoint( - admin_client: DynamicClient, model_registry_instance_service: Service, ) -> str: - return get_endpoint_from_mr_service( - client=admin_client, svc=model_registry_instance_service, protocol=Protocols.REST - ) + return get_endpoint_from_mr_service(svc=model_registry_instance_service, protocol=Protocols.REST) @pytest.fixture(scope="class") diff --git a/tests/model_registry/utils.py b/tests/model_registry/utils.py index 4282bbd6c..86341b5d8 100644 --- a/tests/model_registry/utils.py +++ b/tests/model_registry/utils.py @@ -40,7 +40,7 @@ def get_mr_service_by_label(client: DynamicClient, ns: Namespace, mr_instance: M raise ResourceNotFoundError(f"{mr_instance.name} has no Service") -def get_endpoint_from_mr_service(client: DynamicClient, svc: Service, protocol: str) -> str: +def get_endpoint_from_mr_service(svc: Service, protocol: str) -> str: if protocol in (Protocols.REST, Protocols.GRPC): return svc.instance.metadata.annotations[f"{ADDRESS_ANNOTATION_PREFIX}{protocol}"] else: