diff --git a/selectinf/learning/core.py b/selectinf/learning/core.py index 7ad6b4b68..f58f6ad9d 100644 --- a/selectinf/learning/core.py +++ b/selectinf/learning/core.py @@ -20,6 +20,7 @@ from .samplers import (normal_sampler, split_sampler) from .learners import mixture_learner + try: from .keras_fit import keras_fit except ImportError: