fix: integ tests

Author: evakravi

tests/integ/sagemaker/jumpstart/retrieve_uri/test_inference.py

@@ -31,8 +31,8 @@
 
 def test_jumpstart_inference_retrieve_functions(setup):
 
-    model_id, model_version = "catboost-classification-model", "1.0.0"
-    instance_type = "ml.m5.xlarge"
+    model_id, model_version = "catboost-classification-model", "2.1.6"
+    instance_type = "ml.m5.4xlarge"
 
     print("Starting inference...")
 

tests/integ/sagemaker/jumpstart/retrieve_uri/test_transfer_learning.py

@@ -33,7 +33,7 @@
 
 def test_jumpstart_transfer_learning_retrieve_functions(setup):
 
-    model_id, model_version = "huggingface-spc-bert-base-cased", "1.0.0"
+    model_id, model_version = "huggingface-spc-bert-base-cased", "2.0.3"
     training_instance_type = "ml.p3.2xlarge"
     inference_instance_type = "ml.p2.xlarge"