From e2701ab38be5eb9d0fc394dc372feca3d635e004 Mon Sep 17 00:00:00 2001
From: dbczumar <39497902+dbczumar@users.noreply.github.com>
Date: Wed, 1 Aug 2018 16:26:59 -0700
Subject: [PATCH] Add missing mlflow.version import to sagemaker module (#229)

---
 mlflow/sagemaker/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mlflow/sagemaker/__init__.py b/mlflow/sagemaker/__init__.py
index b25414fb8558d..23d16c143b182 100644
--- 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