remove try_import for optional imports

telekom · Jul 5, 2021 · fe31845 · fe31845
1 parent 3daa4d4
commit fe31845
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/hpoflow/optuna_transformers.py b/hpoflow/optuna_transformers.py
@@ -5,16 +5,10 @@
 import logging
 import os
 
-from hpoflow.optuna_mlflow import OptunaMLflow
-from optuna._imports import try_import
-
-with try_import() as _imports:
-    import mlflow
-    import transformers
+import mlflow
+import transformers
 
-# do the check eagerly and not in the constructor
-# because OptunaMLflowCallback inherits from transformers.TrainerCallback
-_imports.check()
+from hpoflow.optuna_mlflow import OptunaMLflow
 
 _logger = logging.getLogger(__name__)