Skip to content

GitLab

Commit cae394c8 authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Adapt import to new structure

parent 4e0f583e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/sagemaker/scripts/tensorflow/run_tf_dist.py
View file @ cae394c8
...@@ -9,7 +9,7 @@ from datasets import load_dataset ...@@ -9,7 +9,7 @@ from datasets import load_dataset
from tqdm import tqdm from tqdm import tqdm
from transformers import AutoTokenizer, TFAutoModelForSequenceClassification 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(): if os.environ.get("SDP_ENABLED") or is_sagemaker_dp_enabled():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment