From d8e7586b6f73327c5ada999d3923f53f34e42418 Mon Sep 17 00:00:00 2001 From: Sachin Varghese Date: Sun, 15 Oct 2023 14:13:18 -0400 Subject: [PATCH] fixing tests Signed-off-by: Sachin Varghese --- runtimes/alibi-explain/tests/conftest.py | 2 +- runtimes/mlflow/tests/rest/conftest.py | 2 +- tests/conftest.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/runtimes/alibi-explain/tests/conftest.py b/runtimes/alibi-explain/tests/conftest.py index 3298272a9..30ba8d6dc 100644 --- a/runtimes/alibi-explain/tests/conftest.py +++ b/runtimes/alibi-explain/tests/conftest.py @@ -127,7 +127,7 @@ async def model_registry( @pytest.fixture def data_plane(settings: Settings, model_registry: MultiModelRegistry) -> DataPlane: - return DataPlane(settings=settings, model_registry=model_registry) + return DataPlane(settings=settings, model_registry=model_registry,response_cache=None) @pytest.fixture diff --git a/runtimes/mlflow/tests/rest/conftest.py b/runtimes/mlflow/tests/rest/conftest.py index 8a5df840c..1a1b02a59 100644 --- a/runtimes/mlflow/tests/rest/conftest.py +++ b/runtimes/mlflow/tests/rest/conftest.py @@ -95,7 +95,7 @@ def settings() -> Settings: @pytest.fixture def data_plane(settings: Settings, model_registry: MultiModelRegistry) -> DataPlane: - return DataPlane(settings=settings, model_registry=model_registry) + return DataPlane(settings=settings, model_registry=model_registry,response_cache=None) @pytest.fixture diff --git a/tests/conftest.py b/tests/conftest.py index 31e825d45..e5bd99afc 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -203,7 +203,7 @@ def data_plane( model_registry: MultiModelRegistry, prometheus_registry: CollectorRegistry, ) -> DataPlane: - return DataPlane(settings=settings, model_registry=model_registry) + return DataPlane(settings=settings, model_registry=model_registry,response_cache=None) @pytest.fixture