From f7f1b6697e192152738497b5356372c128a207af Mon Sep 17 00:00:00 2001
From: Eugene Fedorenko <efedorenko@Eugenes-MBP.guest.corp.microsoft.com>
Date: Wed, 4 Mar 2020 17:59:36 -0800
Subject: [PATCH] fix import

---
 diabetes_regression/training/train_aml.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/diabetes_regression/training/train_aml.py b/diabetes_regression/training/train_aml.py
index 5bf76cb4..18cfb852 100644
--- a/diabetes_regression/training/train_aml.py
+++ b/diabetes_regression/training/train_aml.py
@@ -23,7 +23,8 @@
 ARISING IN ANY WAY OUT OF THE USE OF THE SOFTWARE CODE, EVEN IF ADVISED OF THE
 POSSIBILITY OF SUCH DAMAGE.
 """
-from azureml.core.run import Dataset, Datastore, Run, Workspace
+from azureml.core.run import Run
+from azureml.core import Dataset, Datastore, Workspace
 import os
 import argparse
 import joblib