refactor: Uniform BRANCH for notebooks

Signed-off-by: Oliver Koenig <okoenig@nvidia.com>
NVIDIA · Jul 12, 2024 · dca5359 · dca5359
1 parent 3e2bb21
commit dca5359
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tutorials/asr/Confidence_Ensembles.ipynb b/tutorials/asr/Confidence_Ensembles.ipynb
@@ -39,7 +39,7 @@
     "\n",
     "# option #2: download NeMo repo\n",
     "if 'google.colab' in str(get_ipython()) or not os.path.exists(os.path.join(NEMO_DIR, \"nemo\")):\n",
-    "    BRANCH = \"main\"\n",
+    "    BRANCH = 'main'\n",
     "    !git clone -b $BRANCH https://github.com/NVIDIA/NeMo $WORKSPACE_DIR/NeMo\n",
     "    NEMO_DIR = os.path.join(WORKSPACE_DIR, 'NeMo')\n",
     "\n",

diff --git a/tutorials/nlp/Token_Classification-BioMegatron.ipynb b/tutorials/nlp/Token_Classification-BioMegatron.ipynb
@@ -7,7 +7,7 @@
             "metadata": {},
             "outputs": [],
             "source": [
-                "BRANCH='main'"
+                "BRANCH = 'main'"
             ]
         },
         {

diff --git a/tutorials/nlp/lora.ipynb b/tutorials/nlp/lora.ipynb
@@ -31,7 +31,7 @@
    "outputs": [],
    "source": [
     "%cd /NeMo/tutorials/nlp\n",
-    "BRANCH='main'\n",
+    "BRANCH = 'main'\n",
     "import os\n",
     "import wget\n",
     "import sys\n",