diff --git a/tests/model_hub_tests/models_hub_common/constants.py b/tests/model_hub_tests/models_hub_common/constants.py index d46c858912d..b00833424d9 100644 --- a/tests/model_hub_tests/models_hub_common/constants.py +++ b/tests/model_hub_tests/models_hub_common/constants.py @@ -22,6 +22,7 @@ tf_hub_cache_dir = os.environ.get('TFHUB_CACHE_DIR', os.path.join(tempfile.gettempdir(), "tfhub_modules")) hf_cache_dir = os.environ.get('HF_HUB_CACHE', os.path.join(tempfile.gettempdir(), "hugging_face")) +# update environment variables if temporal directory is used os.environ['TFHUB_CACHE_DIR'] = tf_hub_cache_dir os.environ['HF_HUB_CACHE'] = hf_cache_dir diff --git a/tests/model_hub_tests/tensorflow/test_tf_convert_model.py b/tests/model_hub_tests/tensorflow/test_tf_convert_model.py index 34651e4632d..fb007db059c 100644 --- a/tests/model_hub_tests/tensorflow/test_tf_convert_model.py +++ b/tests/model_hub_tests/tensorflow/test_tf_convert_model.py @@ -13,7 +13,6 @@ import tensorflow.compat.v1 as tf_v1 import tensorflow_hub as hub # noinspection PyUnresolvedReferences import tensorflow_text # do not delete, needed for text models -from models_hub_common.constants import tf_hub_cache_dir, hf_cache_dir from models_hub_common.test_convert_model import TestConvertModel from models_hub_common.utils import get_models_list from openvino import Core diff --git a/tests/model_hub_tests/tensorflow/test_tf_read_model.py b/tests/model_hub_tests/tensorflow/test_tf_read_model.py index a91b391b2ca..879164e4a05 100644 --- a/tests/model_hub_tests/tensorflow/test_tf_read_model.py +++ b/tests/model_hub_tests/tensorflow/test_tf_read_model.py @@ -10,7 +10,6 @@ import tensorflow as tf import tensorflow_hub as hub # noinspection PyUnresolvedReferences import tensorflow_text # do not delete, needed for text models -from models_hub_common.constants import tf_hub_cache_dir, hf_cache_dir from models_hub_common.test_convert_model import TestConvertModel from models_hub_common.utils import get_models_list from openvino import Core, PartialShape @@ -126,12 +125,6 @@ class TestTFReadModel(TestConvertModel): pass def teardown_method(self): - # remove all downloaded files for TF Hub models - self._clean_dir(tf_hub_cache_dir) - - # remove all downloaded files for HF models - self._clean_dir(hf_cache_dir) - # deallocate memory after each test case gc.collect()