remove imports in init

leftshiftone · Aug 10, 2020 · 721d042 · 721d042
1 parent 6c60887
commit 721d042
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ktrain/__init__.py b/ktrain/__init__.py
@@ -1,5 +1,5 @@
 from .version import __version__
-from . import imports as I
+# from . import imports as I
 from .core import ArrayLearner, GenLearner, get_predictor, load_predictor, release_gpu_memory
 from .vision.learner import ImageClassLearner
 from .text.learner import BERTTextClassLearner, TransformerTextClassLearner
@@ -47,6 +47,7 @@ def get_learner(model, train_data=None, val_data=None,
                                 Only supported for models from vision.image_classifiers
                                 at this time.
     """
+    from . import imports as I
 
     # check arguments
     if not isinstance(model, I.Model):