Adapt import to new structure

2022-03-24 14:40:05 -04:00
parent 4e0f583eea
commit cae394c8fa
1 changed files with 1 additions and 1 deletions
--- a/tests/sagemaker/scripts/tensorflow/run_tf_dist.py
+++ b/tests/sagemaker/scripts/tensorflow/run_tf_dist.py
@@ -9,7 +9,7 @@ from datasets import load_dataset
 from tqdm import tqdm

 from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
-from transformers.file_utils import is_sagemaker_dp_enabled
+from transformers.utils import is_sagemaker_dp_enabled


 if os.environ.get("SDP_ENABLED") or is_sagemaker_dp_enabled():