@@ -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 (
0 commit comments