From cae394c8faac229d43df7559476cfbdd745f8c41 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Thu, 24 Mar 2022 14:40:05 -0400 Subject: [PATCH] Adapt import to new structure --- tests/sagemaker/scripts/tensorflow/run_tf_dist.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/sagemaker/scripts/tensorflow/run_tf_dist.py b/tests/sagemaker/scripts/tensorflow/run_tf_dist.py index 4ff709d037..84f4275aaf 100644 --- 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():