Skip to content

Commit 53b4238

Browse files
authored
fix: XLA tests updated (#1413)
1 parent 7bd9eb5 commit 53b4238

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

examples/notebooks/XLA.ipynb

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -723,7 +723,11 @@
723723
"metadata": {},
724724
"outputs": [],
725725
"source": [
726-
"! cd catalyst && pip install -e ."
726+
"! export GIT_REPO=\"catalyst-team/catalyst\" && git clone https://github.com/${GIT_REPO}\n",
727+
"! cd catalyst && git checkout -b $BRANCH origin/$BRANCH && git pull origin $BRANCH\n",
728+
"! pip install \\\n",
729+
" -r ./catalyst/requirements/requirements.txt \\\n",
730+
" -r ./catalyst/requirements/requirements-dev.txt"
727731
]
728732
},
729733
{

examples/notebooks/XLA_ddp.ipynb

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1066,7 +1066,11 @@
10661066
"metadata": {},
10671067
"outputs": [],
10681068
"source": [
1069-
"! cd catalyst && pip install -e ."
1069+
"! export GIT_REPO=\"catalyst-team/catalyst\" && git clone https://github.com/${GIT_REPO}\n",
1070+
"! cd catalyst && git checkout -b $BRANCH origin/$BRANCH && git pull origin $BRANCH\n",
1071+
"! pip install \\\n",
1072+
" -r ./catalyst/requirements/requirements.txt \\\n",
1073+
" -r ./catalyst/requirements/requirements-dev.txt"
10701074
]
10711075
},
10721076
{

0 commit comments

Comments
 (0)