Update brush.py

HumanSignal · Mar 23, 2023 · ed0bda4 · ed0bda4
1 parent 982067c
commit ed0bda4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/label_studio_converter/brush.py b/label_studio_converter/brush.py
@@ -36,15 +36,19 @@
 from collections import defaultdict
 from itertools import groupby
 
+logger = logging.getLogger(__name__)
+
 try:
     import pycocotools.mask
 except ImportError:
     pycocotools_imported = False
+    logger.warning('pycocotools.mask is not imported, BrushLabels export is not working,'
+                   'you can try to install it manually: pip install pycocotools')
 else:
     pycocotools_imported = True
 
 
-logger = logging.getLogger(__name__)
+
 
 
 ### Brush Export ###