From 2d8f937913fb4451191204ebc9fda6abd03cd37d Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Thu, 9 Mar 2023 15:59:45 -0800 Subject: [PATCH] Improve docstring of image classifier model spec. PiperOrigin-RevId: 515466722 --- .../model_maker/python/vision/image_classifier/model_spec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mediapipe/model_maker/python/vision/image_classifier/model_spec.py b/mediapipe/model_maker/python/vision/image_classifier/model_spec.py index ef44f86e6..d46cafe6b 100644 --- a/mediapipe/model_maker/python/vision/image_classifier/model_spec.py +++ b/mediapipe/model_maker/python/vision/image_classifier/model_spec.py @@ -28,7 +28,7 @@ class ModelSpec(object): uri: str, input_image_shape: Optional[List[int]] = None, name: str = ''): - """Initializes a new instance of the `ImageModelSpec` class. + """Initializes a new instance of the image classifier `ModelSpec` class. Args: uri: str, URI to the pretrained model.