diff --git a/Cargo.lock b/Cargo.lock
index 410cd7534d543..1e5b08d6098e3 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2432,7 +2432,7 @@ dependencies = [
 [[package]]
 name = "pubgrub"
 version = "0.2.1"
-source = "git+https://github.com/zanieb/pubgrub?rev=b5ead05c954b81690aec40255a1c36ec248e90af#b5ead05c954b81690aec40255a1c36ec248e90af"
+source = "git+https://github.com/astral-sh/pubgrub?rev=addbaf184891d66a2dfd93d241a66d13bfe5de86#addbaf184891d66a2dfd93d241a66d13bfe5de86"
 dependencies = [
  "indexmap 2.2.5",
  "log",
diff --git a/Cargo.toml b/Cargo.toml
index c738478769c8b..07395adb1a550 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -67,7 +67,7 @@ owo-colors = { version = "4.0.0" }
 pathdiff = { version = "0.2.1" }
 petgraph = { version = "0.6.4" }
 platform-info = { version = "2.0.2" }
-pubgrub = { git = "https://github.com/zanieb/pubgrub", rev = "b5ead05c954b81690aec40255a1c36ec248e90af" }
+pubgrub = { git = "https://github.com/astral-sh/pubgrub", rev = "addbaf184891d66a2dfd93d241a66d13bfe5de86" }
 pyo3 = { version = "0.20.3" }
 pyo3-log = { version = "0.9.0" }
 pyproject-toml = { version = "0.10.0" }
diff --git a/scripts/requirements/bio_embeddings.in b/scripts/requirements/bio_embeddings.in
new file mode 100644
index 0000000000000..0164d1e72b92c
--- /dev/null
+++ b/scripts/requirements/bio_embeddings.in
@@ -0,0 +1 @@
+bio_embeddings[all]