From 721d042041463a4ce3567a2a8fa4fa975c6b0466 Mon Sep 17 00:00:00 2001 From: Christoph Minixhofer Date: Mon, 10 Aug 2020 10:11:18 +0200 Subject: [PATCH] remove imports in init --- ktrain/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ktrain/__init__.py b/ktrain/__init__.py index 590a420ab..fe1e83fdc 100644 --- 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):