diff --git a/tests/models/test_model.py b/tests/models/test_model.py index ad9d42aa..bd90d5d0 100644 --- a/tests/models/test_model.py +++ b/tests/models/test_model.py @@ -4,7 +4,7 @@ SRC_DIRS = [ - os.path.join(os.path.dirname(__file__), "..", "transformers/examples/pytorch", dirname) + os.path.join("/transformers/examples/pytorch", dirname) for dirname in [ "text-generation", "text-classification",