We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6bc5835 commit c4f3397Copy full SHA for c4f3397
tests/integ/test_huggingface_torch_distributed.py
@@ -15,7 +15,7 @@
15
import os
16
import pytest
17
import sagemaker.utils
18
-from sagemaker.pytorch import HuggingFace
+from sagemaker.huggingface import HuggingFace
19
from tests.integ import timeout
20
from tests.integ import DATA_DIR, TRAINING_DEFAULT_TIMEOUT_MINUTES
21
from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
0 commit comments