diff --git a/test/unit_tests/service_estimator_test.py b/test/unit_tests/service_estimator_test.py index aee2727..7503bd5 100644 --- a/test/unit_tests/service_estimator_test.py +++ b/test/unit_tests/service_estimator_test.py @@ -20,6 +20,8 @@ def input_size(): @pytest.fixture(scope="session") def keras_model(input_size): + import warnings + warnings.filterwarnings("ignore", category=DeprecationWarning) import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'