From 278470233347506a88eab6ee649c11557b7f1bc4 Mon Sep 17 00:00:00 2001
From: Theo Brown <7982453+theo-brown@users.noreply.github.com>
Date: Mon, 24 Mar 2025 12:53:49 +0000
Subject: [PATCH] Update cola version to prepare for jax>0.5.0

---
 pyproject.toml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 2f254558..1a1dbbe5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -28,12 +28,15 @@ dependencies = [
   "tqdm>4.66.2",
   "tensorflow-probability>=0.24.0",
   "beartype>0.16.1",
-  "cola-ml==0.0.5",
+  "cola-ml>=0.0.7",
   "jaxopt==0.8.2",
   "flax<0.10.0",
   "numpy<2.0.0",
 ]
 
+[tool.hatch.metadata]
+allow-direct-references = true
+
 [tool.hatch.envs.default]
 installer = "uv"
 python = "3.10"