From 53b4238d9ff63145cc174ba9a8f052080f84d312 Mon Sep 17 00:00:00 2001
From: Yauheni Kachan <19803638+bagxi@users.noreply.github.com>
Date: Sun, 17 Apr 2022 17:40:36 +0300
Subject: [PATCH] fix: XLA tests updated (#1413)

---
 examples/notebooks/XLA.ipynb     | 6 +++++-
 examples/notebooks/XLA_ddp.ipynb | 6 +++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/examples/notebooks/XLA.ipynb b/examples/notebooks/XLA.ipynb
index ee8ec19af..2a5e0a3b2 100644
--- a/examples/notebooks/XLA.ipynb
+++ b/examples/notebooks/XLA.ipynb
@@ -723,7 +723,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "! cd catalyst && pip install -e ."
+    "! export GIT_REPO=\"catalyst-team/catalyst\" && git clone https://github.com/${GIT_REPO}\n",
+    "! cd catalyst && git checkout -b $BRANCH origin/$BRANCH  && git pull origin $BRANCH\n",
+    "! pip install \\\n",
+    "  -r ./catalyst/requirements/requirements.txt \\\n",
+    "  -r ./catalyst/requirements/requirements-dev.txt"
    ]
   },
   {
diff --git a/examples/notebooks/XLA_ddp.ipynb b/examples/notebooks/XLA_ddp.ipynb
index 837e61741..d329a3ce0 100644
--- a/examples/notebooks/XLA_ddp.ipynb
+++ b/examples/notebooks/XLA_ddp.ipynb
@@ -1066,7 +1066,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "! cd catalyst && pip install -e ."
+    "! export GIT_REPO=\"catalyst-team/catalyst\" && git clone https://github.com/${GIT_REPO}\n",
+    "! cd catalyst && git checkout -b $BRANCH origin/$BRANCH  && git pull origin $BRANCH\n",
+    "! pip install \\\n",
+    "  -r ./catalyst/requirements/requirements.txt \\\n",
+    "  -r ./catalyst/requirements/requirements-dev.txt"
    ]
   },
   {