Skip to content

Commit ecbffbf

Browse files
authored
fix: Add tagging assert to inference recommender integ tests (#3924)
Co-authored-by: Gary Wang 😤 <[email protected]>
1 parent a060a9c commit ecbffbf

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

tests/unit/sagemaker/inference_recommender/test_inference_recommender_mixin.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -585,6 +585,7 @@ def test_deploy_right_size_with_model_package_succeeds(
585585
IR_MODEL_PACKAGE_CONTAINER_DEF,
586586
vpc_config=None,
587587
enable_network_isolation=False,
588+
tags=None,
588589
)
589590

590591
sagemaker_session.endpoint_from_production_variants.assert_called_with(
@@ -615,6 +616,7 @@ def test_deploy_right_size_with_both_overrides_succeeds(
615616
IR_MODEL_PACKAGE_CONTAINER_DEF,
616617
vpc_config=None,
617618
enable_network_isolation=False,
619+
tags=None,
618620
)
619621

620622
sagemaker_session.endpoint_from_production_variants.assert_called_with(
@@ -671,6 +673,7 @@ def test_deploy_right_size_serverless_override(sagemaker_session, default_right_
671673
IR_MODEL_PACKAGE_CONTAINER_DEF,
672674
vpc_config=None,
673675
enable_network_isolation=False,
676+
tags=None,
674677
)
675678

676679
sagemaker_session.endpoint_from_production_variants.assert_called_with(
@@ -703,6 +706,7 @@ def test_deploy_right_size_async_override(sagemaker_session, default_right_sized
703706
IR_MODEL_PACKAGE_CONTAINER_DEF,
704707
vpc_config=None,
705708
enable_network_isolation=False,
709+
tags=None,
706710
)
707711

708712
sagemaker_session.endpoint_from_production_variants.assert_called_with(
@@ -745,6 +749,7 @@ def test_deploy_right_size_explainer_config_override(sagemaker_session, default_
745749
IR_MODEL_PACKAGE_CONTAINER_DEF,
746750
vpc_config=None,
747751
enable_network_isolation=False,
752+
tags=None,
748753
)
749754

750755
sagemaker_session.endpoint_from_production_variants.assert_called_with(

tests/unit/sagemaker/model/test_deploy.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -647,6 +647,7 @@ def test_deploy_with_recommendation_id_with_model_pkg_arn(name_from_base, sagema
647647
IR_MODEL_PACKAGE_CONTAINER_DEF,
648648
vpc_config=None,
649649
enable_network_isolation=False,
650+
tags=None,
650651
)
651652

652653
assert model_package.model_package_arn == IR_MODEL_PACKAGE_VERSION_ARN
@@ -715,6 +716,7 @@ def test_deploy_with_recommendation_id_with_model_pkg_arn_and_compilation(
715716
IR_COMPILATION_MODEL_PACKAGE_CONTAINER_DEF,
716717
vpc_config=None,
717718
enable_network_isolation=False,
719+
tags=None,
718720
)
719721

720722
assert model_package.model_data == IR_COMPILATION_MODEL_DATA

0 commit comments

Comments
 (0)