Skip to content

Commit c91cda5

Browse files
makungaj1Jonathan Makunga
and
Jonathan Makunga
committed
Fix fetch instance rate bug (aws#4624)
Co-authored-by: Jonathan Makunga <[email protected]>
1 parent 76263f5 commit c91cda5

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

src/sagemaker/jumpstart/model.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -922,12 +922,13 @@ def _convert_to_deployment_config_metadata(
922922
instance_rate = get_instance_rate_per_hour(
923923
instance_type=default_inference_instance_type, region=self.region
924924
)
925-
instance_rate_metric = JumpStartBenchmarkStat(instance_rate)
925+
if instance_rate is not None:
926+
instance_rate_metric = JumpStartBenchmarkStat(instance_rate)
926927

927-
if benchmark_metrics is None:
928-
benchmark_metrics = [instance_rate_metric]
929-
else:
930-
benchmark_metrics.append(instance_rate_metric)
928+
if benchmark_metrics is None:
929+
benchmark_metrics = [instance_rate_metric]
930+
else:
931+
benchmark_metrics.append(instance_rate_metric)
931932

932933
init_kwargs = get_init_kwargs(
933934
model_id=self.model_id,

0 commit comments

Comments
 (0)