From f9ed05dd03d2269feb44a4eae41f82463853ff74 Mon Sep 17 00:00:00 2001 From: rasmi Date: Wed, 28 Aug 2024 07:08:29 -0400 Subject: [PATCH] Fix import paths for test_module (#32888) * Fix import path for test_feature_extraction_utils.py See https://github.com/huggingface/transformers/pull/32601 * Fix import path for test_image_processing_utils.py --- tests/utils/test_feature_extraction_utils.py | 2 +- tests/utils/test_image_processing_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/utils/test_feature_extraction_utils.py b/tests/utils/test_feature_extraction_utils.py index d365b4abe9..0d4e4cfb48 100644 --- a/tests/utils/test_feature_extraction_utils.py +++ b/tests/utils/test_feature_extraction_utils.py @@ -27,7 +27,7 @@ from transformers import AutoFeatureExtractor, Wav2Vec2FeatureExtractor from transformers.testing_utils import TOKEN, USER, get_tests_dir, is_staging_test -sys.path.append(str(Path(__file__).parent.parent / "utils")) +sys.path.append(str(Path(__file__).parent.parent.parent / "utils")) from test_module.custom_feature_extraction import CustomFeatureExtractor # noqa E402 diff --git a/tests/utils/test_image_processing_utils.py b/tests/utils/test_image_processing_utils.py index 5cae3095d8..c64dd94ec3 100644 --- a/tests/utils/test_image_processing_utils.py +++ b/tests/utils/test_image_processing_utils.py @@ -27,7 +27,7 @@ from transformers.image_processing_utils import get_size_dict from transformers.testing_utils import TOKEN, USER, get_tests_dir, is_staging_test -sys.path.append(str(Path(__file__).parent.parent / "utils")) +sys.path.append(str(Path(__file__).parent.parent.parent / "utils")) from test_module.custom_image_processing import CustomImageProcessor # noqa E402