Skip to content

Commit 78ee819

Browse files
revert unwanted changes
1 parent 4f507f7 commit 78ee819

File tree

2 files changed

+0
-44
lines changed

2 files changed

+0
-44
lines changed

src/sagemaker/image_uri_config/autogluon.json

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -489,15 +489,12 @@
489489
"cn-north-1": "727897471807",
490490
"cn-northwest-1": "727897471807",
491491
"eu-central-1": "763104351884",
492-
"eu-central-2": "380420809688",
493492
"eu-north-1": "763104351884",
494493
"eu-west-1": "763104351884",
495494
"eu-west-2": "763104351884",
496495
"eu-west-3": "763104351884",
497496
"eu-south-1": "692866216735",
498-
"eu-south-2": "503227376785",
499497
"me-south-1": "217643126080",
500-
"me-central-1": "914824155844",
501498
"sa-east-1": "763104351884",
502499
"us-east-1": "763104351884",
503500
"us-east-2": "763104351884",

tests/unit/test_session.py

Lines changed: 0 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -1234,47 +1234,6 @@ def assert_create_tuning_job_request(**kwrags):
12341234
)
12351235

12361236

1237-
def test_tune_with_strategy_config(sagemaker_session):
1238-
def assert_create_tuning_job_request(**kwrags):
1239-
assert (
1240-
kwrags["HyperParameterTuningJobConfig"]["StrategyConfig"]["HyperbandStrategyConfig"][
1241-
"MinResource"
1242-
]
1243-
== SAMPLE_HYPERBAND_STRATEGY_CONFIG["HyperbandStrategyConfig"]["MinResource"]
1244-
)
1245-
assert (
1246-
kwrags["HyperParameterTuningJobConfig"]["StrategyConfig"]["HyperbandStrategyConfig"][
1247-
"MaxResource"
1248-
]
1249-
== SAMPLE_HYPERBAND_STRATEGY_CONFIG["HyperbandStrategyConfig"]["MaxResource"]
1250-
)
1251-
1252-
sagemaker_session.sagemaker_client.create_hyper_parameter_tuning_job.side_effect = (
1253-
assert_create_tuning_job_request
1254-
)
1255-
sagemaker_session.tune(
1256-
job_name="dummy-tuning-1",
1257-
strategy="Bayesian",
1258-
objective_type="Maximize",
1259-
objective_metric_name="val-score",
1260-
max_jobs=100,
1261-
max_parallel_jobs=5,
1262-
parameter_ranges=SAMPLE_PARAM_RANGES,
1263-
static_hyperparameters=STATIC_HPs,
1264-
image_uri="dummy-image-1",
1265-
input_mode="File",
1266-
metric_definitions=SAMPLE_METRIC_DEF,
1267-
role=EXPANDED_ROLE,
1268-
input_config=SAMPLE_INPUT,
1269-
output_config=SAMPLE_OUTPUT,
1270-
resource_config=RESOURCE_CONFIG,
1271-
stop_condition=SAMPLE_STOPPING_CONDITION,
1272-
tags=None,
1273-
warm_start_config=None,
1274-
strategy_config=SAMPLE_HYPERBAND_STRATEGY_CONFIG,
1275-
)
1276-
1277-
12781237
def test_tune_with_encryption_flag(sagemaker_session):
12791238
def assert_create_tuning_job_request(**kwrags):
12801239
assert (

0 commit comments

Comments
 (0)