diff --git a/google/cloud/vision/entity.py b/google/cloud/vision/entity.py index d456abfe06409..3c8f810c37c77 100644 --- a/google/cloud/vision/entity.py +++ b/google/cloud/vision/entity.py @@ -55,7 +55,7 @@ def from_api_repr(cls, response): :rtype: :class:`~google.cloud.vision.entiy.EntityAnnotation` :returns: Instance of ``EntityAnnotation``. """ - bounds = Bounds.from_api_repr(response.get('boundingPoly', {})) + bounds = Bounds.from_api_repr(response.get('boundingPoly')) description = response['description'] locations = [LocationInformation.from_api_repr(location) for location in response.get('locations', [])] diff --git a/google/cloud/vision/geometry.py b/google/cloud/vision/geometry.py index f32ed61ecfc0d..4e6af390213d6 100644 --- a/google/cloud/vision/geometry.py +++ b/google/cloud/vision/geometry.py @@ -31,9 +31,12 @@ def from_api_repr(cls, response_vertices): :type response_vertices: dict :param response_vertices: List of vertices. - :rtype: :class:`~google.cloud.vision.geometry.BoundsBase` - :returns: Instance of BoundsBase with populated verticies. + :rtype: :class:`~google.cloud.vision.geometry.BoundsBase` or None + :returns: Instance of BoundsBase with populated verticies or None. """ + if not response_vertices: + return None + vertices = [Vertex(vertex.get('x', None), vertex.get('y', None)) for vertex in response_vertices.get('vertices', [])] return cls(vertices)