Skip to content

Commit 099535f

Browse files
committed
fix types
1 parent 785ce0b commit 099535f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

huggingface-sagemaker/steps/promotion/promote_metric_compare_promoter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@
2828

2929
@step
3030
def promote_metric_compare_promoter(
31-
latest_metrics: Dict[str, str],
32-
current_metrics: Dict[str, str],
31+
latest_metrics: Dict[str, float],
32+
current_metrics: Dict[str, float],
3333
metric_to_compare: str = "accuracy",
3434
):
3535
"""Try to promote trained model.

0 commit comments

Comments
 (0)