Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

suppress tf-internal deprication warning

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-25 14:56:47 +01:00
parent 7834a65ff5
commit 31591bef0f
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/unit_tests/service_estimator_test.py
View File

@ -20,6 +20,8 @@ def input_size():
@pytest.fixture(scope="session") @pytest.fixture(scope="session")
def keras_model(input_size): def keras_model(input_size):
import warnings
warnings.filterwarnings("ignore", category=DeprecationWarning)
import os import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'