suppress iobes warnings

OnlpLab · Sep 5, 2021 · f4448f2 · f4448f2
1 parent 2dd0ac6
commit f4448f2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/api_main.py b/api_main.py
@@ -15,8 +15,9 @@
 from operator import itemgetter
 from itertools import groupby
 import iobes
-from logging import ERROR
-iobes.LOGGER.setLevel(level=ERROR)
+if SUPPRESS_IOBES_WARNINGS:
+    from logging import ERROR
+    iobes.LOGGER.setLevel(level=ERROR)
 
 os.environ['CUDA_VISIBLE_DEVICES'] = ''
 

diff --git a/config.py b/config.py
@@ -28,4 +28,6 @@
 
 YAP_API_HEADERS = {'content-type': 'application/json'}
 
-MAX_THREADS_FASTAPI = 5
+MAX_THREADS_FASTAPI = 5
+
+SUPPRESS_IOBES_WARNINGS = False