Add Optional[] type annotation to create_for_inference() methods PiperOrigin-RevId: 355460484
diff --git a/tensorflow_lite_support/metadata/python/metadata_writers/image_classifier.py b/tensorflow_lite_support/metadata/python/metadata_writers/image_classifier.py index 08efb50..90cf860 100644 --- a/tensorflow_lite_support/metadata/python/metadata_writers/image_classifier.py +++ b/tensorflow_lite_support/metadata/python/metadata_writers/image_classifier.py
@@ -83,8 +83,8 @@ @classmethod def create_for_inference(cls, model_buffer: bytearray, - input_norm_mean: List[float], - input_norm_std: List[float], + input_norm_mean: Optional[List[float]], + input_norm_std: Optional[List[float]], label_file_paths: List[str]): """Creates mandatory metadata for TFLite Support inference.
diff --git a/tensorflow_lite_support/metadata/python/metadata_writers/image_segmenter.py b/tensorflow_lite_support/metadata/python/metadata_writers/image_segmenter.py index 1780465..db25413 100644 --- a/tensorflow_lite_support/metadata/python/metadata_writers/image_segmenter.py +++ b/tensorflow_lite_support/metadata/python/metadata_writers/image_segmenter.py
@@ -111,8 +111,8 @@ @classmethod def create_for_inference(cls, model_buffer: bytearray, - input_norm_mean: List[float], - input_norm_std: List[float], + input_norm_mean: Optional[List[float]], + input_norm_std: Optional[List[float]], label_file_paths: List[str]): """Creates mandatory metadata for TFLite Support inference.
diff --git a/tensorflow_lite_support/metadata/python/metadata_writers/object_detector.py b/tensorflow_lite_support/metadata/python/metadata_writers/object_detector.py index 7afb95c..5b50ac0 100644 --- a/tensorflow_lite_support/metadata/python/metadata_writers/object_detector.py +++ b/tensorflow_lite_support/metadata/python/metadata_writers/object_detector.py
@@ -178,8 +178,8 @@ @classmethod def create_for_inference(cls, model_buffer: bytearray, - input_norm_mean: List[float], - input_norm_std: List[float], + input_norm_mean: Optional[List[float]], + input_norm_std: Optional[List[float]], label_file_paths: List[str]): """Creates mandatory metadata for TFLite Support inference.