replace .removesuffix()

Author: benieric

tests/unit/sagemaker/image_uris/test_djl.py

@@ -23,7 +23,7 @@
 )
 def test_djl_uris(load_config_and_file_name):
     config, file_name = load_config_and_file_name
-    framework = file_name.removesuffix(".json")
+    framework = file_name.split(".json")[0]
     VERSIONS = config["versions"]
     for version in VERSIONS:
         ACCOUNTS = config["versions"][version]["registries"]

tests/unit/sagemaker/image_uris/test_dlc_frameworks.py

@@ -61,7 +61,7 @@ def test_chainer_uris(load_config):
 @pytest.mark.parametrize("scope", ["training", "inference", "eia"])
 def test_dlc_framework_uris(load_config_and_file_name, scope):
     config, file_name = load_config_and_file_name
-    framework = file_name.removesuffix(".json")
+    framework = file_name.split(".json")[0]
     VERSIONS = config[scope]["versions"]
 
     for version in VERSIONS:
@@ -115,7 +115,7 @@ def test_dlc_framework_uris(load_config_and_file_name, scope):
 )
 def test_uncommon_format_dlc_framework_version_uris(load_config_and_file_name):
     config, file_name = load_config_and_file_name
-    framework = file_name.removesuffix(".json")
+    framework = file_name.split(".json")[0]
     py_versions = ["py2", "py3"]
 
     # These versions are formatted differently than others for their framework

tests/unit/sagemaker/image_uris/test_graviton.py

@@ -45,7 +45,7 @@ def _test_graviton_framework_uris(framework, version, py_version, account, regio
 @pytest.mark.parametrize("scope", ["inference_graviton"])
 def test_graviton_framework_uris(load_config_and_file_name, scope):
     config, file_name = load_config_and_file_name
-    framework = file_name.removesuffix(".json")
+    framework = file_name.split(".json")[0]
     VERSIONS = config[scope]["versions"]
     for version in VERSIONS:
         ACCOUNTS = config[scope]["versions"][version]["registries"]

tests/unit/sagemaker/image_uris/test_neo.py

@@ -33,7 +33,7 @@
 @pytest.mark.parametrize("load_config_and_file_name", NEO_ALGOS, indirect=True)
 def test_neo_alogos(load_config_and_file_name):
     config, file_name = load_config_and_file_name
-    algo = file_name.removesuffix(".json")
+    algo = file_name.split(".json")[0]
     VERSIONS = config["versions"]
     for version in VERSIONS:
         ACCOUNTS = config["versions"][version]["registries"]
@@ -46,7 +46,7 @@ def test_neo_alogos(load_config_and_file_name):
 @pytest.mark.parametrize("load_config_and_file_name", NEO_AND_INFERENTIA_FRAMEWORKS, indirect=True)
 def test_neo_and_inferentia_frameworks(load_config_and_file_name):
     config, file_name = load_config_and_file_name
-    framework = file_name.removesuffix(".json")
+    framework = file_name.split(".json")[0]
     VERSIONS = config["versions"]
     processors = config["processors"]
     for version in VERSIONS:

tests/unit/sagemaker/image_uris/test_rl.py

@@ -33,7 +33,7 @@
 )
 def test_rl_image_uris(load_config_and_file_name):
     config, filename = load_config_and_file_name
-    framework = filename.removesuffix(".json")
+    framework = filename.split(".json")[0]
     VERSIONS = config["versions"]
     processors = config["processors"]
     for version in VERSIONS: