remove model loading (Azure#1975)

Co-authored-by: Miseon Park <[email protected]>
vrxmike · Jan 6, 2023 · 9c88c15 · 9c88c15
1 parent eb876ce
commit 9c88c15
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 12 deletions.
diff --git a/...automl-with-azureml/automl-nlp-multiclass/automl-nlp-text-classification-multiclass.ipynb b/...automl-with-azureml/automl-nlp-multiclass/automl-nlp-text-classification-multiclass.ipynb
@@ -391,7 +391,7 @@
    "outputs": [],
    "source": [
     "automl_run = experiment.submit(automl_config, show_output=False)\n",
-    "_ = automl_run.wait_for_completion(show_output=False)"
+    "automl_run.wait_for_completion(show_output=False)"
    ]
   },
   {
@@ -428,7 +428,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "best_run, best_model = automl_run.get_output()\n",
+    "best_run = automl_run.get_best_child()\n",
     "best_run"
    ]
   },

diff --git a/...automl-with-azureml/automl-nlp-multilabel/automl-nlp-text-classification-multilabel.ipynb b/...automl-with-azureml/automl-nlp-multilabel/automl-nlp-text-classification-multilabel.ipynb
@@ -71,7 +71,6 @@
     "import pandas as pd\n",
     "from sklearn.metrics import classification_report\n",
     "\n",
-    "from azureml.automl.dnn.nlp.common.io.utils import load_model_wrapper\n",
     "from azureml.automl.runtime.shared.score.scoring import score_classification\n",
     "import azureml.core\n",
     "from azureml.core import Dataset\n",
@@ -292,7 +291,7 @@
     "automl_run = experiment.submit(\n",
     "    automl_config, show_output=False\n",
     ")  # You might see a warning about \"enable_distributed_dnn_training\". Please simply ignore.\n",
-    "_ = automl_run.wait_for_completion(show_output=False)"
+    "automl_run.wait_for_completion(show_output=False)"
    ]
   },
   {
@@ -329,12 +328,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "(\n",
-    "    best_run,\n",
-    "    best_model,\n",
-    ") = (\n",
-    "    automl_run.get_output()\n",
-    ")  # You might see a warning about \"enable_distributed_dnn_training\". Please simply ignore.\n",
+    "best_run = automl_run.get_best_child()\n",
     "best_run"
    ]
   },
@@ -482,6 +476,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from azureml.automl.dnn.nlp.common.io.utils import load_model_wrapper\n",
     "y_transformer = load_model_wrapper(training_run).y_transformer"
    ]
   },

diff --git a/v1/python-sdk/tutorials/automl-with-azureml/automl-nlp-ner/automl-nlp-ner.ipynb b/v1/python-sdk/tutorials/automl-with-azureml/automl-nlp-ner/automl-nlp-ner.ipynb
@@ -272,7 +272,7 @@
    "outputs": [],
    "source": [
     "automl_run = experiment.submit(automl_config, show_output=False)\n",
-    "_ = automl_run.wait_for_completion(show_output=False)"
+    "automl_run.wait_for_completion(show_output=False)"
    ]
   },
   {
@@ -309,7 +309,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "best_run, best_model = automl_run.get_output()\n",
+    "best_run = automl_run.get_best_child()\n",
     "best_run"
    ]
   },