Error message on shapely package absence

Author: hovnatan

.vscode/settings.json

@@ -1,4 +1,4 @@
 {
     "python.pythonPath": "~/miniconda3/bin/python",
     "python.formatting.provider": "yapf"
-}
+}

install.sh

@@ -11,6 +11,7 @@ rm -rf venv_sa_conv
 python${PYTHON_VER} -m venv venv_sa_conv
 source venv_sa_conv/bin/activate
 
+pip install shapely
 pip install -e .
 # pip install --pre superannotate
 # pip install superannotate

superannotate/__init__.py

@@ -22,7 +22,17 @@
     from .consensus_benchmark.benchmark import benchmark
     from .consensus_benchmark.consensus import consensus
 except:
-    pass
+
+    def benchmark(*args, **kwargs):
+        raise RuntimeError(
+            "To use superannotate.benchmark or superannotate.consensus functions please install shapely package with # conda install shapely or # pip install shapely"
+        )
+
+    def consensus(*args, **kwargs):
+        raise RuntimeError(
+            "To use superannotate.benchmark or superannotate.consensus functions please install shapely package with # conda install shapely or # pip install shapely"
+        )
+
 
 from .dataframe_filtering import (
     filter_annotation_instances, filter_images_by_comments,

superannotate/version.py

@@ -1 +1 @@
-__version__ = "2.5.9"
+__version__ = "2.5.10"

tests/consensus_benchmark/test_benchmark.py

@@ -1,6 +1,6 @@
 from pathlib import Path
 
-from superannotate.consensus_benchmark.benchmark import benchmark
+import superannotate as sa
 
 test_root = Path().resolve() / 'tests'
 
@@ -15,7 +15,7 @@ def test_benchmark():
     ]
     export_path = test_root / 'consensus_benchmark'
     for annot_type in annot_types:
-        res_df = benchmark(
+        res_df = sa.benchmark(
             gt_project_name,
             project_names,
             export_root=export_path,
@@ -39,7 +39,7 @@ def test_benchmark():
     ]
 
     #test filtering images with given image names list
-    res_images = benchmark(
+    res_images = sa.benchmark(
         gt_project_name,
         project_names,
         export_root=export_path,

tests/consensus_benchmark/test_consensus.py

@@ -1,6 +1,6 @@
 from pathlib import Path
 
-from superannotate.consensus_benchmark.consensus import consensus
+import superannotate as sa
 
 test_root = Path().resolve() / 'tests'
 
@@ -14,7 +14,7 @@ def test_consensus():
     ]
     export_path = test_root / 'consensus_benchmark'
     for annot_type in annot_types:
-        res_df = consensus(
+        res_df = sa.consensus(
             project_names, export_root=export_path, annot_type=annot_type
         )
         #test content of projectName column
@@ -35,7 +35,7 @@ def test_consensus():
     ]
 
     #test filtering images with given image names list
-    res_images = consensus(
+    res_images = sa.consensus(
         project_names, export_root=export_path, image_list=image_names
     )