Skip to content

Commit 6fb935d

Browse files
makungaj1Jonathan Makunga
and
Jonathan Makunga
authored
ModelBuilder to support display with filter. (#4712)
Co-authored-by: Jonathan Makunga <[email protected]>
1 parent 2be02e0 commit 6fb935d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/serve/builder/jumpstart_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -458,12 +458,12 @@ def get_deployment_config(self) -> Optional[Dict[str, Any]]:
458458

459459
return self.pysdk_model.deployment_config
460460

461-
def display_benchmark_metrics(self):
461+
def display_benchmark_metrics(self, **kwargs):
462462
"""Display Markdown Benchmark Metrics for deployment configs."""
463463
if not hasattr(self, "pysdk_model") or self.pysdk_model is None:
464464
self._build_for_jumpstart()
465465

466-
self.pysdk_model.display_benchmark_metrics()
466+
self.pysdk_model.display_benchmark_metrics(**kwargs)
467467

468468
def list_deployment_configs(self) -> List[Dict[str, Any]]:
469469
"""List deployment configs for ``This`` model in the current region.

0 commit comments

Comments
 (0)