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" ] }, {