Addressed PR Review comments

Author: Jonathan Makunga

src/sagemaker/serve/builder/jumpstart_builder.py

@@ -42,7 +42,7 @@
 )
 from sagemaker.serve.utils.telemetry_logger import _capture_telemetry
 from sagemaker.serve.utils.tuning import (
-    _pretty_print_benchmark_results,
+    _pretty_print_results_jumpstart,
     _serial_benchmark,
     _concurrent_benchmark,
     _more_performant,
@@ -386,7 +386,7 @@ def _tune_for_js(self, max_tuning_duration: int = 1800):
         if best_tuned_combination:
             self.pysdk_model.env.update({num_shard_env_var_name: str(best_tuned_combination[1])})
 
-            _pretty_print_benchmark_results(benchmark_results, [num_shard_env_var_name])
+            _pretty_print_results_jumpstart(benchmark_results, [num_shard_env_var_name])
             logger.info(
                 "Model Configuration: %s was most performant with avg latency: %s, "
                 "p90 latency: %s, average tokens per second: %s, throughput/s: %s, "

src/sagemaker/serve/utils/tuning.py

@@ -99,7 +99,7 @@ def _pretty_print_results_tgi(results: dict):
     )
 
 
-def _pretty_print_benchmark_results(results: dict, model_env_vars=None):
+def _pretty_print_results_jumpstart(results: dict, model_env_vars=None):
     """Pretty prints benchmark results"""
     if model_env_vars is None:
         model_env_vars = []