From 724856cc69d95be9aa30d3167d07d72b76b7a760 Mon Sep 17 00:00:00 2001 From: brandon Date: Tue, 9 Apr 2024 14:56:08 -0700 Subject: [PATCH] Makes the SDK communicate over the wire with the proper label value string --- src/groundlight/binary_labels.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/groundlight/binary_labels.py b/src/groundlight/binary_labels.py index de3216a1..4d8bd923 100644 --- a/src/groundlight/binary_labels.py +++ b/src/groundlight/binary_labels.py @@ -74,9 +74,7 @@ def convert_display_label_to_internal( if not isinstance(label, str): raise ValueError(f"Expected a string label, but got {label} of type {type(label)}") upper = label.upper() - if upper == Label.YES: - return DeprecatedLabel.PASS.value - if upper == Label.NO: - return DeprecatedLabel.FAIL.value + if upper in {Label.YES.value, Label.NO.value}: + return upper raise ValueError(f"Invalid label string '{label}'. Must be one of '{Label.YES.value}','{Label.NO.value}'.")