Update notebook

AgnostiqHQ · Sep 30, 2023 · c4b01e1 · c4b01e1
1 parent ece2070
commit c4b01e1
Showing 1 changed file with 14 additions and 17 deletions.
diff --git a/doc/source/tutorials/0_ClassicalMachineLearning/genai/source.ipynb b/doc/source/tutorials/0_ClassicalMachineLearning/genai/source.ipynb
@@ -45,7 +45,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 2,
    "id": "03a0c441-40ec-4353-82eb-4d395fb8e4e2",
    "metadata": {},
    "outputs": [
@@ -75,7 +75,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 3,
    "id": "3846e49b-f0fa-4878-b733-5fa53c1452af",
    "metadata": {},
    "outputs": [],
@@ -86,7 +86,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 6,
    "id": "063f6e62-402f-4232-b6e2-f94568aee059",
    "metadata": {},
    "outputs": [],
@@ -106,7 +106,8 @@
     "from diffusers import DiffusionPipeline\n",
     "from PIL import Image, ImageDraw, ImageFont\n",
     "import covalent as ct\n",
-    "import torch"
+    "import torch\n",
+    "import logging"
    ]
   },
   {
@@ -122,24 +123,20 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 7,
    "id": "13658ad0-c002-4ef6-bc4d-ad9fed565443",
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[2023-09-30 06:24:07,120] [DEBUG] azurebatch.py: Line 183 in _debug_log: Azure Batch Executor: Starting Azure Batch Executor with config:\n",
-      "[2023-09-30 06:24:07,120] [DEBUG] azurebatch.py: Line 183 in _debug_log: Azure Batch Executor: {'tenant_id': '9494b93c-f5da-4db4-985e-29b5ef769e2e', 'client_id': '8a1005e8-6271-4832-8350-710564d20d81', 'client_secret': 'iJS8Q~Y-kGbyd49wb2kstAkVP3e4L6RMv-82Ga~N', 'batch_account_url': 'https://blogcovalentbatch.eastus.batch.azure.com', 'batch_account_domain': 'batch.core.windows.net', 'storage_account_name': 'blogcovalentbatch', 'storage_account_domain': 'blob.core.windows.net', 'base_image_uri': 'blogcovalentbatch.azurecr.io/covalent-executor-base:latest', 'pool_id': 'covalent-cpu', 'retries': 3, 'time_limit': 600, 'cache_dir': '/tmp/covalent', 'poll_freq': 10}\n",
-      "[2023-09-30 06:24:07,144] [DEBUG] azurebatch.py: Line 183 in _debug_log: Azure Batch Executor: Starting Azure Batch Executor with config:\n",
-      "[2023-09-30 06:24:07,145] [DEBUG] azurebatch.py: Line 183 in _debug_log: Azure Batch Executor: {'tenant_id': '9494b93c-f5da-4db4-985e-29b5ef769e2e', 'client_id': '8a1005e8-6271-4832-8350-710564d20d81', 'client_secret': 'iJS8Q~Y-kGbyd49wb2kstAkVP3e4L6RMv-82Ga~N', 'batch_account_url': 'https://blogcovalentbatch.eastus.batch.azure.com', 'batch_account_domain': 'batch.core.windows.net', 'storage_account_name': 'blogcovalentbatch', 'storage_account_domain': 'blob.core.windows.net', 'base_image_uri': 'docker.io/filipbolt/covalent_azure:0.220.0', 'pool_id': 'covalent-gpu', 'retries': 3, 'time_limit': 600, 'cache_dir': '/tmp/covalent', 'poll_freq': 10}\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "# save under workflow.py\n",
     "\n",
+    "# setting loggers to info to avoid too many debug messages\n",
+    "loggers = [logging.getLogger(name) for name in logging.root.manager.loggerDict]\n",
+    "for logger in loggers:\n",
+    "    logger.setLevel(logging.INFO)\n",
+    "\n",
+    "\n",
+    "\n",
     "# define dependencies to install on remote execution\n",
     "DEPS_ALL = ct.DepsPip(\n",
     "    packages=[\n",