update model package tests

gwang111 · gwang111 · commit 5ec2f4aaa5a0 · 2023-06-07T09:02:05.000-07:00
diff --git a/tests/unit/sagemaker/model/test_model_package.py b/tests/unit/sagemaker/model/test_model_package.py
@@ -115,6 +115,7 @@ def test_create_sagemaker_model_uses_model_name(name_from_base, sagemaker_sessio
         {"ModelPackageName": model_package_name},
         vpc_config=None,
         enable_network_isolation=False,
+        tags=None,
     )
 
 
@@ -141,6 +142,7 @@ def test_create_sagemaker_model_include_environment_variable(sagemaker_session):
         {"ModelPackageName": model_package_name, "Environment": environment},
         vpc_config=None,
         enable_network_isolation=False,
+        tags=None,
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -115,6 +115,7 @@ def test_create_sagemaker_model_uses_model_name(name_from_base, sagemaker_sessio`
`115`	`115`	`{"ModelPackageName": model_package_name},`
`116`	`116`	`vpc_config=None,`
`117`	`117`	`enable_network_isolation=False,`
	`118`	`+ tags=None,`
`118`	`119`	`)`
`119`	`120`
`120`	`121`
`@@ -141,6 +142,7 @@ def test_create_sagemaker_model_include_environment_variable(sagemaker_session):`
`141`	`142`	`{"ModelPackageName": model_package_name, "Environment": environment},`
`142`	`143`	`vpc_config=None,`
`143`	`144`	`enable_network_isolation=False,`
	`145`	`+ tags=None,`
`144`	`146`	`)`
`145`	`147`
`146`	`148`