Skip to content

Commit cc99a57

Browse files
committed
chore: add unit test for Model.create() with eula
1 parent 2a8d685 commit cc99a57

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

tests/unit/sagemaker/model/test_model.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -322,6 +322,21 @@ def test_create_sagemaker_model_accelerator_type(prepare_container_def, sagemake
322322
)
323323

324324

325+
@patch("sagemaker.model.Model.prepare_container_def")
326+
def test_create_sagemaker_model_with_eula(prepare_container_def, sagemaker_session):
327+
model = Model(MODEL_IMAGE, MODEL_DATA, name=MODEL_NAME, sagemaker_session=sagemaker_session)
328+
329+
accelerator_type = "ml.eia.medium"
330+
model.create(INSTANCE_TYPE, accelerator_type=accelerator_type, accept_eula=True)
331+
332+
prepare_container_def.assert_called_with(
333+
INSTANCE_TYPE,
334+
accelerator_type=accelerator_type,
335+
serverless_inference_config=None,
336+
accept_eula=True,
337+
)
338+
339+
325340
@patch("sagemaker.model.Model.prepare_container_def")
326341
def test_create_sagemaker_model_tags(prepare_container_def, sagemaker_session):
327342
container_def = {"Image": MODEL_IMAGE, "Environment": {}, "ModelDataUrl": MODEL_DATA}

0 commit comments

Comments
 (0)