Backwards compatibility with IBEIS

WildMeOrg · Jul 15, 2020 · d168dcd · d168dcd
1 parent 2e383e8
commit d168dcd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/wbia_pie/_plugin.py b/wbia_pie/_plugin.py
@@ -27,7 +27,10 @@
 (print, rrr, profile) = ut.inject2(__name__)
 
 _, register_ibs_method = controller_inject.make_ibs_register_decorator(__name__)
-register_api = controller_inject.get_wbia_flask_api(__name__)
+if USE_WBIA:
+    register_api = controller_inject.get_wbia_flask_api(__name__)
+else:
+    register_api = controller_inject.get_ibeis_flask_api(__name__)
 register_preproc_annot = controller_inject.register_preprocs['annot']
 
 _PLUGIN_FOLDER = os.path.dirname(os.path.realpath(__file__))