(fix)(demos): modified the shell script to download the binaries befo…

…re running the demo notebooks
ivy-llc · Jun 25, 2024 · d92f9c6 · d92f9c6
1 parent a8bf755
commit d92f9c6
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 10 deletions.
diff --git a/examples_and_demos/alexnet_demo_cpu.ipynb b/examples_and_demos/alexnet_demo_cpu.ipynb
@@ -123,9 +123,7 @@
         "!git clone https://github.com/unifyai/models.git --depth 1\n",
         "\n",
         "# Installing models package from cloned repository! 😄\n",
-        "!cd models/ && pip install .\n",
-        "\n",
-        "exit()"
+        "!cd models/ && pip install .\n"
       ]
     },
     {

diff --git a/examples_and_demos/image_segmentation_with_ivy_unet_cpu.ipynb b/examples_and_demos/image_segmentation_with_ivy_unet_cpu.ipynb
@@ -37,9 +37,7 @@
         "!git clone https://github.com/unifyai/models.git --depth 1\n",
         "\n",
         "# Installing models package from cloned repository! 😄\n",
-        "!cd models/ && pip install .\n",
-        "\n",
-        "exit()"
+        "!cd models/ && pip install .\n"
       ]
     },
     {

diff --git a/examples_and_demos/resnet_demo_cpu.ipynb b/examples_and_demos/resnet_demo_cpu.ipynb
@@ -47,9 +47,7 @@
         "# Installing models package from cloned repository! 😄\n",
         "!cd models/ && pip install .\n",
         "\n",
-        "!python3 -m pip install torchvision\n",
-        "\n",
-        "exit()"
+        "!python3 -m pip install torchvision\n"
       ]
     },
     {

diff --git a/tests/test_demos.sh b/tests/test_demos.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 if [ $2 == true ]; then
-export VERSION=nightly
+export VERSION=main
 fi
 
 cd ivy
@@ -13,6 +13,13 @@ mkdir .ivy
 # install dependencies
 pip install -r requirements.txt >/dev/null 2>&1
 
+# get the binaries
+python3 << 'EOF'
+import ivy
+ivy.utils.cleanup_and_fetch_binaries(clean=True)
+EOF
+
+
 # run test
 if [ $3 == true ]; then
     echo "PATH : $1"