diff --git a/robotoff/prediction/object_detection/utils/label_map_util.py b/robotoff/prediction/object_detection/utils/label_map_util.py index 0b2e84a593..bf0c136528 100644 --- a/robotoff/prediction/object_detection/utils/label_map_util.py +++ b/robotoff/prediction/object_detection/utils/label_map_util.py @@ -141,7 +141,7 @@ def load_labelmap(path: str): """ with open(path, "r") as fid: label_map_string = fid.read() - label_map = string_int_label_map_pb2.StringIntLabelMap() + label_map = string_int_label_map_pb2.StringIntLabelMap() # type: ignore try: text_format.Merge(label_map_string, label_map) except text_format.ParseError: diff --git a/robotoff/prediction/object_detection/utils/string_int_label_map_pb2.py b/robotoff/prediction/object_detection/utils/string_int_label_map_pb2.py index 48d90f6b87..3f1d729da5 100644 --- a/robotoff/prediction/object_detection/utils/string_int_label_map_pb2.py +++ b/robotoff/prediction/object_detection/utils/string_int_label_map_pb2.py @@ -1,5 +1,6 @@ # Generated by the protocol buffer compiler. DO NOT EDIT! # source: robotoff/ml/object_detection/utils/string_int_label_map.proto +# type: ignore import sys