Skip to content

Commit 58bdbd0

Browse files
authored
chore: pin framework version in serverless inference tests (aws#4803)
1 parent a7f1cab commit 58bdbd0

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

tests/integ/test_sklearn.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -159,8 +159,6 @@ def test_deploy_model(
159159
def test_deploy_model_with_serverless_inference_config(
160160
sklearn_training_job,
161161
sagemaker_session,
162-
sklearn_latest_version,
163-
sklearn_latest_py_version,
164162
):
165163
endpoint_name = unique_name_from_base("test-sklearn-deploy-model-serverless")
166164
with timeout_and_delete_endpoint_by_name(endpoint_name, sagemaker_session):
@@ -173,7 +171,7 @@ def test_deploy_model_with_serverless_inference_config(
173171
model_data,
174172
ROLE,
175173
entry_point=script_path,
176-
framework_version=sklearn_latest_version,
174+
framework_version="1.0-1",
177175
sagemaker_session=sagemaker_session,
178176
)
179177
predictor = model.deploy(

tests/integ/test_xgboost.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -121,11 +121,9 @@ def test_training_with_network_isolation(
121121
]
122122

123123

124-
@pytest.mark.skip(reason="re:Invent keynote3 blocker. Revisit after release")
125124
def test_xgboost_serverless_inference(
126125
xgboost_training_job,
127126
sagemaker_session,
128-
xgboost_latest_version,
129127
):
130128
endpoint_name = unique_name_from_base("test-xgboost-deploy-model-serverless")
131129
with timeout_and_delete_endpoint_by_name(endpoint_name, sagemaker_session):
@@ -139,7 +137,7 @@ def test_xgboost_serverless_inference(
139137
model_data=model_data,
140138
role=ROLE,
141139
entry_point=os.path.join(DATA_DIR, "xgboost_abalone", "abalone.py"),
142-
framework_version=xgboost_latest_version,
140+
framework_version="1.5-1",
143141
)
144142

145143
xgboost.deploy(

0 commit comments

Comments
 (0)