Update notebook parameters.

scenarios/entailment/entailment_multinli_bert.ipynb

@@ -46,7 +46,11 @@
    "outputs": [],
    "source": [
     "## Set QUICK_RUN = True to run the notebook on a small subset of data and a smaller number of epochs.\n",
-    "QUICK_RUN = False"
+    "QUICK_RUN = False\n",
+    "\n",
+    "TRAIN_DATA_USED_PERCENT = 1\n",
+    "DEV_DATA_USED_PERCENT = 1\n",
+    "NUM_EPOCHS = 2"
    ]
   },
   {
@@ -59,10 +63,6 @@
     "    TRAIN_DATA_USED_PERCENT = 0.001\n",
     "    DEV_DATA_USED_PERCENT = 0.01\n",
     "    NUM_EPOCHS = 1\n",
-    "else:\n",
-    "    TRAIN_DATA_USED_PERCENT = 1\n",
-    "    DEV_DATA_USED_PERCENT = 1\n",
-    "    NUM_EPOCHS = 2\n",
     "\n",
     "import torch\n",
     "if torch.cuda.is_available():\n",

tests/integration/test_notebooks_entailment_multinli_bert.py

@@ -12,6 +12,8 @@ def test_entailment_multinli_bert(notebooks):
     pm.execute_notebook(
         notebook_path,
         OUTPUT_NOTEBOOK,
-        parameters={"QUICK_RUN": True},
+        parameters={"TRAIN_DATA_USED_PERCENT": 0.001, 
+                    "DEV_DATA_USED_PERCENT": 0.01,
+                    "NUM_EPOCHS": 1},
         kernel_name=KERNEL_NAME,
     )