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 2020from sagemaker .jumpstart .hub .hub import Hub
2121
2222from sagemaker .jumpstart .estimator import JumpStartEstimator
23+ from sagemaker .jumpstart .utils import get_jumpstart_content_bucket
24+
2325from tests .integ .sagemaker .jumpstart .constants import (
2426 ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME ,
2527 ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID ,
2931 get_public_hub_model_arn ,
3032 get_sm_session ,
3133 with_exponential_backoff ,
32- )
33- from tests .integ .sagemaker .jumpstart .utils import (
34- get_sm_session ,
3534 get_training_dataset_for_model_and_version ,
3635)
3736
38- from sagemaker .jumpstart .utils import get_jumpstart_content_bucket
39-
4037MAX_INIT_TIME_SECONDS = 5
4138
4239TEST_MODEL_IDS = {
You can’t perform that action at this time.
0 commit comments