fix import

microsoft · Jun 22, 2020 · 1ca7a59 · 1ca7a59
1 parent 743e301
commit 1ca7a59
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/diabetes_regression/evaluate/evaluate_model.py b/diabetes_regression/evaluate/evaluate_model.py
@@ -26,7 +26,7 @@
 from azureml.core import Run
 import argparse
 import traceback
-from util.model_helper import get_latest_model
+from util.model_helper import get_model
 
 run = Run.get_context()
 
@@ -45,7 +45,7 @@
 #         sources_dir = 'diabetes_regression'
 #     path_to_util = os.path.join(".", sources_dir, "util")
 #     sys.path.append(os.path.abspath(path_to_util))  # NOQA: E402
-#     from model_helper import get_latest_model
+#     from model_helper import get_model
 #     workspace_name = os.environ.get("WORKSPACE_NAME")
 #     experiment_name = os.environ.get("EXPERIMENT_NAME")
 #     resource_group = os.environ.get("RESOURCE_GROUP")
@@ -108,7 +108,7 @@
     firstRegistration = False
     tag_name = 'experiment_name'
 
-    model = get_latest_model(
+    model = get_model(
                 model_name=model_name,
                 tag_name=tag_name,
                 tag_value=exp.name,