From 47590a737ed96b685a0d3b2b0bc99b30fcbec16d Mon Sep 17 00:00:00 2001
From: Tyler Cox <tyler.a.cox@berkeley.edu>
Date: Mon, 5 Feb 2024 13:25:38 -0800
Subject: [PATCH] package typo

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 226d64f85..ce0ddd30c 100644
--- a/setup.py
+++ b/setup.py
@@ -63,7 +63,7 @@ def package_files(package_dir, subdirectory):
         "rich",
         "jax",
         "jaxlib",
-        "optax=0.1.7",
+        "optax==0.1.7",
         'attrs',
         'frozendict',
     ],