Refactoring

Author: Jonathan Makunga

src/sagemaker/jumpstart/utils.py

@@ -1220,8 +1220,6 @@ def get_metrics_from_deployment_configs(
     if not deployment_configs:
         return {}
 
-    print("deployment_configs: {}".format(deployment_configs))
-
     data = {"Instance Type": [], "Config Name": [], "Concurrent Users": []}
     instance_rate_data = {}
     for index, deployment_config in enumerate(deployment_configs):

src/sagemaker/utils.py

@@ -1791,9 +1791,9 @@ def extract_instance_rate_per_hour(price_data: Dict[str, Any]) -> Optional[Dict[
                     if value is not None:
                         value = str(round(float(value), 3))
                     return {
-                        "unit": f"{currency}/{price.get('unit', 'Hrs')}",
+                        "unit": f"{currency}/Hr",
                         "value": value,
-                        "name": "Instance Rate",
+                        "name": "On-demand Instance Rate",
                     }
     return None
 

tests/unit/test_utils.py

@@ -1929,7 +1929,7 @@ def test_deep_override_skip_keys(self):
                     }
                 ]
             },
-            {"name": "Instance Rate", "unit": "USD/Hrs", "value": "0.9"},
+            {"name": "On-demand Instance Rate", "unit": "USD/Hr", "value": "0.9"},
         ),
         (
             "ml.t4g.nano",
@@ -1947,7 +1947,7 @@ def test_deep_override_skip_keys(self):
                     '"termAttributes": {}}}}}'
                 ]
             },
-            {"name": "Instance Rate", "unit": "USD/Hrs", "value": "0.008"},
+            {"name": "On-demand Instance Rate", "unit": "USD/Hr", "value": "0.008"},
         ),
         (
             "ml.t4g.nano",
@@ -1965,7 +1965,7 @@ def test_deep_override_skip_keys(self):
                     '"termAttributes": {}}}}}'
                 ]
             },
-            {"name": "Instance Rate", "unit": "USD/Hrs", "value": "0.008"},
+            {"name": "On-demand Instance Rate", "unit": "USD/Hr", "value": "0.008"},
         ),
         (
             "ml.t4g.nano",
@@ -1983,7 +1983,7 @@ def test_deep_override_skip_keys(self):
                     '"termAttributes": {}}}}}'
                 ]
             },
-            {"name": "Instance Rate", "unit": "USD/Hrs", "value": "0.008"},
+            {"name": "On-demand Instance Rate", "unit": "USD/Hr", "value": "0.008"},
         ),
     ],
 )
@@ -2024,7 +2024,7 @@ def test_get_instance_rate_per_hour(
                     }
                 }
             },
-            {"name": "Instance Rate", "unit": "USD/Hrs", "value": "0.9"},
+            {"name": "On-demand Instance Rate", "unit": "USD/Hr", "value": "0.9"},
         ),
     ],
 )