From 06bc347c9795149b89b803e4909cce9a803b3907 Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Date: Mon, 10 Aug 2020 11:16:17 -0400
Subject: [PATCH] Fix links for open in colab (#6391)

---
 docs/source/_static/js/custom.js | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/docs/source/_static/js/custom.js b/docs/source/_static/js/custom.js
index 0aeec4f6ec..9838a10ef4 100644
--- a/docs/source/_static/js/custom.js
+++ b/docs/source/_static/js/custom.js
@@ -98,13 +98,14 @@ function addColabLink() {
     const pageName = parts[parts.length - 1].split(".")[0];
 
     if (hasNotebook.includes(pageName)) {
+        const baseURL = "https://colab.research.google.com/github/huggingface/notebooks/blob/master/transformers_doc/"
         const linksColab = `
         <div class="colab-dropdown">
             <img alt="Open In Colab" src="https://colab.research.google.com/assets/colab-badge.svg">
             <div class="colab-dropdown-content">
-                <button onclick=" window.open('https://colab.research.google.com/github/sgugger/test_nbs/blob/master/${pageName}.ipynb')">Mixed</button>
-                <button onclick=" window.open('https://colab.research.google.com/github/sgugger/test_nbs/blob/master/pytorch/${pageName}.ipynb')">PyTorch</button>
-                <button onclick=" window.open('https://colab.research.google.com/github/sgugger/test_nbs/blob/master/tensorflow/${pageName}.ipynb')">TensorFlow</button>
+                <button onclick=" window.open('${baseURL}${pageName}.ipynb')">Mixed</button>
+                <button onclick=" window.open('${baseURL}pytorch/${pageName}.ipynb')">PyTorch</button>
+                <button onclick=" window.open('${baseURL}tensorflow/${pageName}.ipynb')">TensorFlow</button>
             </div>
         </div>`
         const leftMenu = document.querySelector(".wy-breadcrumbs-aside")