Merge pull request #267 from superannotateai/friday

fix shapely import

Author: dshabin

requirements.txt

@@ -14,7 +14,6 @@ pandas>=1.1.4
 plotly==4.1.0
 ffmpeg-python>=0.2.0
 fire==0.4.0
-Shapely==1.7.1
 mixpanel==4.8.3
 pydantic>=1.8.2
 pydantic[email]

src/superannotate/lib/app/analytics/common.py

@@ -6,10 +6,6 @@
 import plotly.express as px
 from lib.app.exceptions import AppException
 from lib.core import DEPRICATED_DOCUMENT_VIDEO_MESSAGE
-from shapely.geometry import box
-from shapely.geometry import Point
-from shapely.geometry import Polygon
-
 
 logger = logging.getLogger()
 
@@ -542,6 +538,17 @@ def image_consensus(df, image_name, annot_type):
     :type dataset_format: str
 
     """
+
+    try:
+        from shapely.geometry import box
+        from shapely.geometry import Point
+        from shapely.geometry import Polygon
+    except ImportError:
+        raise ImportError(
+            "To use superannotate.benchmark or superannotate.consensus functions please install "
+            "shapely package in Anaconda enviornment with # conda install shapely"
+        )
+
     image_df = df[df["imageName"] == image_name]
     all_projects = list(set(df["folderName"]))
     column_names = [