From 1c89efd607140ed63218e893894728f31508dac7 Mon Sep 17 00:00:00 2001
From: Bryce Meyer <bryce13950@gmail.com>
Date: Sat, 28 Dec 2024 01:00:19 +0100
Subject: [PATCH] removed jaxtyping ceiling

---
 poetry.lock    | 16 +++++++---------
 pyproject.toml |  2 +-
 2 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index 0277aae1a..b45cdee6c 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1508,19 +1508,17 @@ requirements-deprecated-finder = ["pip-api", "pipreqs"]
 
 [[package]]
 name = "jaxtyping"
-version = "0.2.19"
-description = "Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees."
+version = "0.2.36"
+description = "Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays."
 optional = false
-python-versions = "~=3.8"
+python-versions = ">=3.9"
 files = [
-    {file = "jaxtyping-0.2.19-py3-none-any.whl", hash = "sha256:651352032799d422987e783fd1b77699b53c3bb28ffa644bbca5f75ec4fbb843"},
-    {file = "jaxtyping-0.2.19.tar.gz", hash = "sha256:21ff4c3caec6781cadfe980b019dde856c1011e17d11dfe8589298040056325a"},
+    {file = "jaxtyping-0.2.36-py3-none-any.whl", hash = "sha256:b19bcbd4009df8734602203402483a4066ad2eb3382904432e370588e9c9707d"},
+    {file = "jaxtyping-0.2.36.tar.gz", hash = "sha256:781ac44a3cf8982063d7ee48b5008ccfad7b13793bf878eb3058d5319aa08f0f"},
 ]
 
 [package.dependencies]
-numpy = ">=1.20.0"
-typeguard = ">=2.13.3"
-typing-extensions = ">=3.7.4.1"
+typing-extensions = {version = "*", markers = "python_version < \"3.10\""}
 
 [[package]]
 name = "jedi"
@@ -5775,4 +5773,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<4.0"
-content-hash = "b7f3409e0e8c1a97bd054d5c02647aff8786c1fade4b12e72be27e4e9affd78f"
+content-hash = "4de54b517fd98067b63e2a8a07af5577c9c830a929ffce72c72a044477cb009c"
diff --git a/pyproject.toml b/pyproject.toml
index 5ec6d31b2..80f18cf08 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,7 +19,7 @@
         datasets=">=2.7.1"
         einops=">=0.6.0"
         fancy-einsum=">=0.0.3"
-        jaxtyping=">=0.2.11,<0.2.20"
+        jaxtyping=">=0.2.11"
         numpy=[
             {version=">=1.24,<2.0", python=">=3.9,<3.12"},
             {version=">=1.26", python=">=3.12,<3.13"},