From e502ea9570b003f408fdd308871ed1de24b0a3b9 Mon Sep 17 00:00:00 2001
From: Priyansi <priyanhiman@gmail.com>
Date: Fri, 19 Nov 2021 14:28:15 +0530
Subject: [PATCH] fix version extraction

---
 examples/notebooks/MNIST_on_TPU.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/notebooks/MNIST_on_TPU.ipynb b/examples/notebooks/MNIST_on_TPU.ipynb
index 3896a7ad14bc..4e3ff33bb0a3 100644
--- a/examples/notebooks/MNIST_on_TPU.ipynb
+++ b/examples/notebooks/MNIST_on_TPU.ipynb
@@ -68,7 +68,7 @@
       },
       "source": [
         "VERSION = !curl -s https://api.github.com/repos/pytorch/xla/releases/latest | grep -Po '\"tag_name\": \"v\\K.*?(?=\")'\n",
-        "VERSION = VERSION[0].rstrip('.0') # remove trailing zero\n",
+        "VERSION = VERSION[0].rstrip('0').rstrip('.') # remove trailing zero\n",
         "!pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-{VERSION}-cp37-cp37m-linux_x86_64.whl"
       ],
       "execution_count": null,