Skip to content

Commit 68edb38

Browse files
philschmidclaytonparnell
authored andcommitted
Update test_huggingface_torch_distributed.py
1 parent 6d8ece0 commit 68edb38

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_huggingface_torch_distributed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import os
1616
import pytest
1717
import sagemaker.utils
18-
from sagemaker.pytorch import HuggingFace
18+
from sagemaker.huggingface import HuggingFace
1919
from tests.integ import timeout
2020
from tests.integ import DATA_DIR, TRAINING_DEFAULT_TIMEOUT_MINUTES
2121
from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name

0 commit comments

Comments
 (0)