Add missing mlflow.version import to sagemaker module (mlflow#229)

hongye-sun · Aug 1, 2018 · e2701ab · e2701ab
1 parent 1480e60
commit e2701ab
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mlflow/sagemaker/__init__.py b/mlflow/sagemaker/__init__.py
@@ -10,6 +10,7 @@
 import base64
 import boto3
 import mlflow
+import mlflow.version
 from mlflow import pyfunc
 from mlflow.models import Model
 from mlflow.tracking import _get_model_log_dir