revert: lazy

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Aug 26, 2022 · 02df3af · 02df3af
1 parent 24e44f7
commit 02df3af
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/bentoml/_internal/frameworks/mlflow.py b/bentoml/_internal/frameworks/mlflow.py
@@ -4,6 +4,7 @@
 import shutil
 import typing as t
 import logging
+import tempfile
 from typing import TYPE_CHECKING
 
 import bentoml
@@ -185,8 +186,6 @@ def import_model(
         metadata=metadata,
         context=context,
     ) as bento_model:
-        import tempfile
-
         from mlflow.models import Model as MLflowModel
         from mlflow.pyfunc import FLAVOR_NAME as PYFUNC_FLAVOR_NAME
         from mlflow.models.model import MLMODEL_FILE_NAME