File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
tests/integ/sagemaker/jumpstart/private_hub/estimator Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change 20
20
from sagemaker .jumpstart .hub .hub import Hub
21
21
22
22
from sagemaker .jumpstart .estimator import JumpStartEstimator
23
+ from sagemaker .jumpstart .utils import get_jumpstart_content_bucket
24
+
23
25
from tests .integ .sagemaker .jumpstart .constants import (
24
26
ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME ,
25
27
ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID ,
29
31
get_public_hub_model_arn ,
30
32
get_sm_session ,
31
33
with_exponential_backoff ,
32
- )
33
- from tests .integ .sagemaker .jumpstart .utils import (
34
- get_sm_session ,
35
34
get_training_dataset_for_model_and_version ,
36
35
)
37
36
38
- from sagemaker .jumpstart .utils import get_jumpstart_content_bucket
39
-
40
37
MAX_INIT_TIME_SECONDS = 5
41
38
42
39
TEST_MODEL_IDS = {
You can’t perform that action at this time.
0 commit comments