Skip to content

Commit 5c59d54

Browse files
authored
Merge branch 'master' into baselinefix
2 parents 3223ea7 + 4884d18 commit 5c59d54

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

src/sagemaker/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -574,6 +574,7 @@ def _upload_code(self, key_prefix: str, repack: bool = False) -> None:
574574
script=self.entry_point,
575575
directory=self.source_dir,
576576
dependencies=self.dependencies,
577+
kms_key=self.model_kms_key,
577578
settings=self.sagemaker_session.settings,
578579
)
579580

tests/integ/test_inference_recommender.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,6 @@ def test_default_right_size_and_deploy_registered_model_sklearn(
301301
predictor.delete_endpoint()
302302

303303

304-
@pytest.mark.skip(reason="This test is currently failing. Skipping until fixed")
305304
@pytest.mark.slow_test
306305
def test_default_right_size_and_deploy_unregistered_model_sklearn(
307306
default_right_sized_unregistered_model, sagemaker_session
@@ -346,7 +345,6 @@ def test_default_right_size_and_deploy_unregistered_base_model(
346345
predictor.delete_endpoint()
347346

348347

349-
@pytest.mark.skip(reason="This test is currently failing. Skipping until fixed")
350348
@pytest.mark.slow_test
351349
def test_advanced_right_size_and_deploy_unregistered_model_sklearn(
352350
advanced_right_sized_unregistered_model, sagemaker_session

0 commit comments

Comments
 (0)