change: use model_uri estimator class attribute in jumpstart integ tests

evakravi · evakravi · commit f8f7d4b2d993 · 2022-01-24T19:51:11.000Z
diff --git a/tests/integ/sagemaker/jumpstart/script_mode_class/test_transfer_learning.py b/tests/integ/sagemaker/jumpstart/script_mode_class/test_transfer_learning.py
@@ -69,6 +69,7 @@ def test_jumpstart_transfer_learning_estimator_class(setup):
     estimator = Estimator(
         image_uri=image_uri,
         source_dir=script_uri,
+        model_uri=model_uri,
         entry_point=TRAINING_ENTRYPOINT_SCRIPT_NAME,
         role=get_sm_session().get_caller_identity_arn(),
         sagemaker_session=get_sm_session(),
@@ -83,7 +84,6 @@ def test_jumpstart_transfer_learning_estimator_class(setup):
         {
             "training": f"s3://{get_jumpstart_content_bucket(JUMPSTART_DEFAULT_REGION_NAME)}/"
             f"{get_training_dataset_for_model_and_version(model_id, model_version)}",
-            "model": model_uri,
         }
     )
 

Original file line number	Diff line number	Diff line change
`@@ -69,6 +69,7 @@ def test_jumpstart_transfer_learning_estimator_class(setup):`
`69`	`69`	`estimator = Estimator(`
`70`	`70`	`image_uri=image_uri,`
`71`	`71`	`source_dir=script_uri,`
	`72`	`+ model_uri=model_uri,`
`72`	`73`	`entry_point=TRAINING_ENTRYPOINT_SCRIPT_NAME,`
`73`	`74`	`role=get_sm_session().get_caller_identity_arn(),`
`74`	`75`	`sagemaker_session=get_sm_session(),`
`@@ -83,7 +84,6 @@ def test_jumpstart_transfer_learning_estimator_class(setup):`
`83`	`84`	`{`
`84`	`85`	`"training": f"s3://{get_jumpstart_content_bucket(JUMPSTART_DEFAULT_REGION_NAME)}/"`
`85`	`86`	`f"{get_training_dataset_for_model_and_version(model_id, model_version)}",`
`86`		`- "model": model_uri,`
`87`	`87`	`}`
`88`	`88`	`)`
`89`	`89`