From 94a6e122a10b42474741e28e8859e04a9c5d05da Mon Sep 17 00:00:00 2001
From: Du Phan <phandu@google.com>
Date: Wed, 17 Apr 2024 09:01:24 -0400
Subject: [PATCH] fix pypi publish (#32)

---
 .github/workflows/pytest_and_autopublish.yml | 6 ++++--
 pyproject.toml                               | 2 +-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/pytest_and_autopublish.yml b/.github/workflows/pytest_and_autopublish.yml
index 6628dde..986a296 100644
--- a/.github/workflows/pytest_and_autopublish.yml
+++ b/.github/workflows/pytest_and_autopublish.yml
@@ -18,7 +18,7 @@ jobs:
 
     - run: sudo apt install -y pandoc gsfonts
     - run: pip --version
-    - run: pip install -e .[dev,doc,oryx]
+    - run: pip install -e .[dev,doc]
     - run: pip freeze
 
     - name: Run lint
@@ -46,7 +46,9 @@ jobs:
         # cache: pip
         # cache-dependency-path: '**/pyproject.toml'
 
-    - run: pip install -e .[dev,oryx]
+    - run:
+        pip install -e .[dev]
+        pip install "git+https://github.com/jax-ml/oryx.git@b59ab020780cd53d488bc7dcad3696be9fdca0a5"
 
     # Run tests (in parallel)
     - name: Run core tests
diff --git a/pyproject.toml b/pyproject.toml
index ebd65d0..0429e33 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,7 +57,7 @@ doc = [
     "sphinx-gallery",
 ]
 oryx = [
-    "oryx@git+https://github.com/jax-ml/oryx.git@b59ab020780cd53d488bc7dcad3696be9fdca0a5",
+    "oryx",
 ]
 
 [tool.pyink]