From 19e3113556b0306e87e304e04523ba2f0030f6c0 Mon Sep 17 00:00:00 2001
From: Colin Carroll <colcarroll@google.com>
Date: Tue, 19 Dec 2023 11:25:23 -0800
Subject: [PATCH] Update oryx version and add other requirements.

PiperOrigin-RevId: 592294308
---
 pyproject.toml | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index e07f17f..3b03f1a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,12 @@ keywords = []
 dependencies = [
   "jax>=0.4.6",
   "tensorflow-probability[jax]>=0.19.0",
-  "oryx>=0.2.5",
+  "oryx@git+https://github.com/jax-ml/oryx",
+  "arviz",
+  "optax",
+  "blackjax",
+  "numpyro",
+  "jaxopt",
 ]
 
 # `version` is automatically set by flit to use `bayeux.__version__`