Fix Python tutorials for Colab (#503)

ElliottKasoar · web-flow · commit 3f0c13dcef69 · 2025-04-04T14:23:44.000+01:00
diff --git a/docs/source/tutorials/python/eos.ipynb b/docs/source/tutorials/python/eos.ipynb
@@ -38,7 +38,10 @@
    "source": [
     "# import locale\n",
     "# locale.getpreferredencoding = lambda: \"UTF-8\"\n",
-    "# !python3 -m pip install janus-core[all] data-tutorials"
+    "\n",
+    "# ! pip uninstall torch torchaudio torchvision numpy -y\n",
+    "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",
+    "# get_ipython().kernel.do_shutdown(restart=True)"
    ]
   },
   {
diff --git a/docs/source/tutorials/python/geom_opt.ipynb b/docs/source/tutorials/python/geom_opt.ipynb
@@ -38,7 +38,10 @@
    "source": [
     "# import locale\n",
     "# locale.getpreferredencoding = lambda: \"UTF-8\"\n",
-    "# !python3 -m pip install janus-core[all] data-tutorials"
+    "\n",
+    "# ! pip uninstall torch torchaudio torchvision numpy -y\n",
+    "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",
+    "# get_ipython().kernel.do_shutdown(restart=True)"
    ]
   },
   {
diff --git a/docs/source/tutorials/python/md.ipynb b/docs/source/tutorials/python/md.ipynb
@@ -38,7 +38,10 @@
    "source": [
     "# import locale\n",
     "# locale.getpreferredencoding = lambda: \"UTF-8\"\n",
-    "# !python3 -m pip install janus-core[all] data-tutorials"
+    "\n",
+    "# ! pip uninstall torch torchaudio torchvision numpy -y\n",
+    "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",
+    "# get_ipython().kernel.do_shutdown(restart=True)"
    ]
   },
   {
diff --git a/docs/source/tutorials/python/neb.ipynb b/docs/source/tutorials/python/neb.ipynb
@@ -71,7 +71,10 @@
    "source": [
     "# import locale\n",
     "# locale.getpreferredencoding = lambda: \"UTF-8\"\n",
-    "# !python3 -m pip install janus-core[all] data-tutorials"
+    "\n",
+    "# ! pip uninstall torch torchaudio torchvision numpy -y\n",
+    "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",
+    "# get_ipython().kernel.do_shutdown(restart=True)"
    ]
   },
   {
diff --git a/docs/source/tutorials/python/phonons.ipynb b/docs/source/tutorials/python/phonons.ipynb
@@ -38,7 +38,10 @@
    "source": [
     "# import locale\n",
     "# locale.getpreferredencoding = lambda: \"UTF-8\"\n",
-    "# !python3 -m pip install janus-core[all] data-tutorials"
+    "\n",
+    "# ! pip uninstall torch torchaudio torchvision numpy -y\n",
+    "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",
+    "# get_ipython().kernel.do_shutdown(restart=True)"
    ]
   },
   {
diff --git a/docs/source/tutorials/python/single_point.ipynb b/docs/source/tutorials/python/single_point.ipynb
@@ -47,7 +47,10 @@
    "source": [
     "# import locale\n",
     "# locale.getpreferredencoding = lambda: \"UTF-8\"\n",
-    "# !python3 -m pip install janus-core[all] data-tutorials"
+    "\n",
+    "# ! pip uninstall torch torchaudio torchvision numpy -y\n",
+    "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",
+    "# get_ipython().kernel.do_shutdown(restart=True)"
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,10 @@`
`38`	`38`	`"source": [`
`39`	`39`	`"# import locale\n",`
`40`	`40`	`"# locale.getpreferredencoding = lambda: \"UTF-8\"\n",`
`41`		`- "# !python3 -m pip install janus-core[all] data-tutorials"`
	`41`	`+ "\n",`
	`42`	`+ "# ! pip uninstall torch torchaudio torchvision numpy -y\n",`
	`43`	`+ "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",`
	`44`	`+ "# get_ipython().kernel.do_shutdown(restart=True)"`
`42`	`45`	`]`
`43`	`46`	`},`
`44`	`47`	`{`
Original file line number	Diff line number	Diff line change
`@@ -71,7 +71,10 @@`
`71`	`71`	`"source": [`
`72`	`72`	`"# import locale\n",`
`73`	`73`	`"# locale.getpreferredencoding = lambda: \"UTF-8\"\n",`
`74`		`- "# !python3 -m pip install janus-core[all] data-tutorials"`
	`74`	`+ "\n",`
	`75`	`+ "# ! pip uninstall torch torchaudio torchvision numpy -y\n",`
	`76`	`+ "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",`
	`77`	`+ "# get_ipython().kernel.do_shutdown(restart=True)"`
`75`	`78`	`]`
`76`	`79`	`},`
`77`	`80`	`{`
Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,10 @@`
`47`	`47`	`"source": [`
`48`	`48`	`"# import locale\n",`
`49`	`49`	`"# locale.getpreferredencoding = lambda: \"UTF-8\"\n",`
`50`		`- "# !python3 -m pip install janus-core[all] data-tutorials"`
	`50`	`+ "\n",`
	`51`	`+ "# ! pip uninstall torch torchaudio torchvision numpy -y\n",`
	`52`	`+ "# ! uv pip install janus-core[all] data-tutorials torch==2.5.1 --system\n",`
	`53`	`+ "# get_ipython().kernel.do_shutdown(restart=True)"`
`51`	`54`	`]`
`52`	`55`	`},`
`53`	`56`	`{`