suppress tf-internal deprication warning

2022-03-25 14:56:47 +01:00 · 2022-03-25 14:56:47 +01:00 · 31591bef0f
commit 31591bef0f
parent 7834a65ff5
1 changed files with 2 additions and 0 deletions
--- 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'