fix: extra argument

evakravi · evakravi · commit 348f835a2270 · 2023-09-28T19:22:58.000Z
diff --git a/tests/integ/sagemaker/jumpstart/retrieve_uri/test_inference.py b/tests/integ/sagemaker/jumpstart/retrieve_uri/test_inference.py
@@ -62,7 +62,6 @@ def test_jumpstart_inference_retrieve_functions(setup):
         script_uri=script_uri,
         model_uri=model_uri,
         instance_type=instance_type,
-        base_name="catboost",
         environment_variables=environment_vars,
     )
 
diff --git a/tests/integ/sagemaker/jumpstart/retrieve_uri/test_transfer_learning.py b/tests/integ/sagemaker/jumpstart/retrieve_uri/test_transfer_learning.py
@@ -68,7 +68,6 @@ def test_jumpstart_transfer_learning_retrieve_functions(setup):
         hyperparameters=default_hyperparameters,
         instance_type=training_instance_type,
         training_dataset_s3_key=get_training_dataset_for_model_and_version(model_id, model_version),
-        base_name="huggingface",
     )
 
     training_job.create_training_job()
@@ -100,7 +99,6 @@ def test_jumpstart_transfer_learning_retrieve_functions(setup):
             training_job.output_tarball_base_path, training_job.training_job_name
         ),
         instance_type=inference_instance_type,
-        base_name="huggingface",
         environment_variables=environment_vars,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,6 @@ def test_jumpstart_inference_retrieve_functions(setup):`
`62`	`62`	`script_uri=script_uri,`
`63`	`63`	`model_uri=model_uri,`
`64`	`64`	`instance_type=instance_type,`
`65`		`- base_name="catboost",`
`66`	`65`	`environment_variables=environment_vars,`
`67`	`66`	`)`
`68`	`67`
Original file line number	Diff line number	Diff line change
`@@ -68,7 +68,6 @@ def test_jumpstart_transfer_learning_retrieve_functions(setup):`
`68`	`68`	`hyperparameters=default_hyperparameters,`
`69`	`69`	`instance_type=training_instance_type,`
`70`	`70`	`training_dataset_s3_key=get_training_dataset_for_model_and_version(model_id, model_version),`
`71`		`- base_name="huggingface",`
`72`	`71`	`)`
`73`	`72`
`74`	`73`	`training_job.create_training_job()`
`@@ -100,7 +99,6 @@ def test_jumpstart_transfer_learning_retrieve_functions(setup):`
`100`	`99`	`training_job.output_tarball_base_path, training_job.training_job_name`
`101`	`100`	`),`
`102`	`101`	`instance_type=inference_instance_type,`
`103`		`- base_name="huggingface",`
`104`	`102`	`environment_variables=environment_vars,`
`105`	`103`	`)`
`106`	`104`