Merge pull request #437 from Labelbox/al-1518

AL-1518 - update to allow tiled image epsg transformer

Author: jtsodapop

examples/annotation_types/tiled_imagery_basics.ipynb

@@ -27,6 +27,8 @@
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+VectorTool = Union[Point, Line, Rectangle, Polygon]
+
 
 class EPSG(Enum):
     """ Provides the EPSG for tiled image assets that are currently supported.
@@ -355,8 +357,6 @@ def validate_zoom_levels(cls, zoom_levels):
 class EPSGTransformer(BaseModel):
     """Transformer class between different EPSG's. Useful when wanting to project
     in different formats.
-
-    Requires as input a Point object.
     """
 
     class Config:
@@ -509,11 +509,16 @@ def _get_point_obj(self, point) -> Point:
 
     def __call__(
         self, shape: Union[Point, Line, Rectangle, Polygon]
-    ) -> Union[Point, Line, Rectangle, Polygon]:
+    ) -> Union[VectorTool, List[VectorTool]]:
+        if isinstance(shape, list):
+            return [self(geom) for geom in shape]
         if isinstance(shape, Point):
             return self._get_point_obj(shape)
-        if isinstance(shape, Line) or isinstance(shape, Polygon):
+        if isinstance(shape, Line):
             return Line(points=[self._get_point_obj(p) for p in shape.points])
+        if isinstance(shape, Polygon):
+            return Polygon(
+                points=[self._get_point_obj(p) for p in shape.points])
         if isinstance(shape, Rectangle):
             return Rectangle(start=self._get_point_obj(shape.start),
                              end=self._get_point_obj(shape.end))