Use mock arize client for testing

caraml-dev · Jan 25, 2024 · c8c6c0c · c8c6c0c
1 parent 3c738d9
commit c8c6c0c
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/python/observation-publisher/tests/test_observability_backend.py b/python/observation-publisher/tests/test_observability_backend.py
@@ -8,13 +8,7 @@
 
 import pandas as pd
 import pyarrow as pa
-from arize.pandas.validation.errors import ValidationError
-from arize.pandas.validation.validator import Validator
-from arize.utils.types import (
-    ModelTypes as ArizeModelType,
-    Environments as ArizeEnvironment,
-    Schema as ArizeSchema,
-)
+from arize.pandas.logger import Client
 from merlin.observability.inference import (
     BinaryClassificationOutput,
     InferenceSchema,