From 334cc0f3f38e7d82e5c1f10216eba14eb8d0db26 Mon Sep 17 00:00:00 2001
From: hakotesova <148503796+hakotesova@users.noreply.github.com>
Date: Fri, 24 Jan 2025 08:50:21 -0800
Subject: [PATCH] Wait 30s after creating deployment before scoring to avoid
 connection error (#3497)

* try increasing online request settings timeout and queue wait

* reformat
---
 .../automl-classification-task-bankmarketing.ipynb         | 7 ++++---
 ...ml-forecasting-task-energy-demand-advanced-mlflow.ipynb | 5 ++++-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/sdk/python/jobs/automl-standalone-jobs/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb b/sdk/python/jobs/automl-standalone-jobs/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb
index f16a09d578e..398dcfac58a 100644
--- a/sdk/python/jobs/automl-standalone-jobs/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb
+++ b/sdk/python/jobs/automl-standalone-jobs/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb
@@ -730,9 +730,12 @@
    },
    "outputs": [],
    "source": [
+    "import time\n",
+    "\n",
     "# bankmarketing deployment to take 100% traffic\n",
     "endpoint.traffic = {\"bankmarketing-deploy\": 100}\n",
-    "ml_client.begin_create_or_update(endpoint)"
+    "ml_client.begin_create_or_update(endpoint)\n",
+    "time.sleep(30)  # sleep 30 seconds before scoring to avoid connection error"
    ]
   },
   {
@@ -782,8 +785,6 @@
    "outputs": [],
    "source": [
     "# wait and delete endpoint\n",
-    "import time\n",
-    "\n",
     "time.sleep(60)"
    ]
   },
diff --git a/sdk/python/jobs/automl-standalone-jobs/automl-forecasting-task-energy-demand/automl-forecasting-task-energy-demand-advanced-mlflow.ipynb b/sdk/python/jobs/automl-standalone-jobs/automl-forecasting-task-energy-demand/automl-forecasting-task-energy-demand-advanced-mlflow.ipynb
index 6189fbe83e2..d57b98cb784 100644
--- a/sdk/python/jobs/automl-standalone-jobs/automl-forecasting-task-energy-demand/automl-forecasting-task-energy-demand-advanced-mlflow.ipynb
+++ b/sdk/python/jobs/automl-standalone-jobs/automl-forecasting-task-energy-demand/automl-forecasting-task-energy-demand-advanced-mlflow.ipynb
@@ -1124,7 +1124,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "ml_client.online_deployments.begin_create_or_update(deployment).wait()"
+    "import time\n",
+    "\n",
+    "ml_client.online_deployments.begin_create_or_update(deployment).wait()\n",
+    "time.sleep(30)  # wait 30 seconds before scoring to avoid connection error"
    ]
   },
   {