update test

pintaoz · pintaoz · commit 377cf1557c5a · 2025-02-26T16:10:48.000-08:00
diff --git a/tests/unit/sagemaker/model/test_framework_model.py b/tests/unit/sagemaker/model/test_framework_model.py
@@ -518,7 +518,6 @@ def test_is_repack_with_none_type(repack_model, sagemaker_session):
     model = FrameworkModel(
         role=ROLE,
         sagemaker_session=sagemaker_session,
-        source_dir="s3://codebucket/someprefix/sourcedir.tar.gz",
         image_uri=IMAGE_URI,
         model_data=MODEL_DATA,
     )
diff --git a/tests/unit/sagemaker/model/test_model.py b/tests/unit/sagemaker/model/test_model.py
@@ -1053,7 +1053,6 @@ def test_is_repack_with_none_type(repack_model, sagemaker_session):
     model = Model(
         role=ROLE,
         sagemaker_session=sagemaker_session,
-        source_dir=SCRIPT_URI,
         image_uri=IMAGE_URI,
         model_data=MODEL_DATA,
     )

Original file line number	Diff line number	Diff line change
`@@ -518,7 +518,6 @@ def test_is_repack_with_none_type(repack_model, sagemaker_session):`
`518`	`518`	`model = FrameworkModel(`
`519`	`519`	`role=ROLE,`
`520`	`520`	`sagemaker_session=sagemaker_session,`
`521`		`- source_dir="s3://codebucket/someprefix/sourcedir.tar.gz",`
`522`	`521`	`image_uri=IMAGE_URI,`
`523`	`522`	`model_data=MODEL_DATA,`
`524`	`523`	`)`
Original file line number	Diff line number	Diff line change
`@@ -1053,7 +1053,6 @@ def test_is_repack_with_none_type(repack_model, sagemaker_session):`
`1053`	`1053`	`model = Model(`
`1054`	`1054`	`role=ROLE,`
`1055`	`1055`	`sagemaker_session=sagemaker_session,`
`1056`		`- source_dir=SCRIPT_URI,`
`1057`	`1056`	`image_uri=IMAGE_URI,`
`1058`	`1057`	`model_data=MODEL_DATA,`
`1059`	`1058`	`)`