@@ -99,7 +99,6 @@ def test_get_jumpstart_gated_content_bucket_override():
99
99
100
100
101
101
def test_get_jumpstart_launched_regions_message ():
102
-
103
102
with patch ("sagemaker.jumpstart.constants.JUMPSTART_REGION_NAME_SET" , {}):
104
103
assert (
105
104
utils .get_jumpstart_launched_regions_message ()
@@ -147,7 +146,6 @@ def test_get_formatted_manifest():
147
146
148
147
149
148
def test_parse_sagemaker_version ():
150
-
151
149
with patch ("sagemaker.__version__" , "1.2.3" ):
152
150
assert utils .parse_sagemaker_version () == "1.2.3"
153
151
@@ -188,7 +186,6 @@ def test_get_sagemaker_version(patched_parse_sm_version: Mock):
188
186
189
187
190
188
def test_is_jumpstart_model_uri ():
191
-
192
189
assert not utils .is_jumpstart_model_uri ("fdsfdsf" )
193
190
assert not utils .is_jumpstart_model_uri ("s3://not-jumpstart-bucket/sdfsdfds" )
194
191
assert not utils .is_jumpstart_model_uri ("some/actual/localfile" )
@@ -689,7 +686,6 @@ def test_add_jumpstart_uri_tags_training():
689
686
690
687
691
688
def test_update_inference_tags_with_jumpstart_training_script_tags ():
692
-
693
689
random_tag_1 = {"Key" : "tag-key-1" , "Value" : "tag-val-1" }
694
690
random_tag_2 = {"Key" : "tag-key-2" , "Value" : "tag-val-2" }
695
691
@@ -750,7 +746,6 @@ def test_update_inference_tags_with_jumpstart_training_script_tags():
750
746
751
747
752
748
def test_update_inference_tags_with_jumpstart_training_model_tags ():
753
-
754
749
random_tag_1 = {"Key" : "tag-key-1" , "Value" : "tag-val-1" }
755
750
random_tag_2 = {"Key" : "tag-key-2" , "Value" : "tag-val-2" }
756
751
@@ -811,7 +806,6 @@ def test_update_inference_tags_with_jumpstart_training_model_tags():
811
806
812
807
813
808
def test_update_inference_tags_with_jumpstart_training_script_tags_inference ():
814
-
815
809
random_tag_1 = {"Key" : "tag-key-1" , "Value" : "tag-val-1" }
816
810
random_tag_2 = {"Key" : "tag-key-2" , "Value" : "tag-val-2" }
817
811
@@ -872,7 +866,6 @@ def test_update_inference_tags_with_jumpstart_training_script_tags_inference():
872
866
873
867
874
868
def test_update_inference_tags_with_jumpstart_training_model_tags_inference ():
875
-
876
869
random_tag_1 = {"Key" : "tag-key-1" , "Value" : "tag-val-1" }
877
870
random_tag_2 = {"Key" : "tag-key-2" , "Value" : "tag-val-2" }
878
871
@@ -975,7 +968,6 @@ def make_vulnerable_inference_spec(*largs, **kwargs):
975
968
976
969
@patch ("sagemaker.jumpstart.utils.accessors.JumpStartModelsAccessor._get_manifest" )
977
970
def test_jumpstart_old_model_spec (mock_get_manifest ):
978
-
979
971
mock_get_manifest .return_value = [
980
972
JumpStartModelHeader (
981
973
{
@@ -1282,7 +1274,6 @@ def test_validate_model_id_and_get_type_false(
1282
1274
)
1283
1275
1284
1276
with patch ("sagemaker.jumpstart.utils.validate_model_id_and_get_type" , patched ):
1285
-
1286
1277
self .assertFalse (utils .validate_model_id_and_get_type ("dee" ))
1287
1278
self .assertFalse (utils .validate_model_id_and_get_type ("" ))
1288
1279
self .assertFalse (utils .validate_model_id_and_get_type (None ))
@@ -1478,7 +1469,6 @@ class TestJumpStartLogger(TestCase):
1478
1469
@patch ("logging.StreamHandler.emit" )
1479
1470
@patch ("sagemaker.jumpstart.constants.JUMPSTART_LOGGER.propagate" , False )
1480
1471
def test_logger_normal_mode (self , mocked_emit : Mock ):
1481
-
1482
1472
JUMPSTART_LOGGER .warning ("Self destruct in 3...2...1..." )
1483
1473
1484
1474
mocked_emit .assert_called_once ()
@@ -1487,7 +1477,6 @@ def test_logger_normal_mode(self, mocked_emit: Mock):
1487
1477
@patch ("logging.StreamHandler.emit" )
1488
1478
@patch ("sagemaker.jumpstart.constants.JUMPSTART_LOGGER.propagate" , False )
1489
1479
def test_logger_disabled (self , mocked_emit : Mock ):
1490
-
1491
1480
JUMPSTART_LOGGER .warning ("Self destruct in 3...2...1..." )
1492
1481
1493
1482
mocked_emit .assert_not_called ()
0 commit comments