From 62496b6972366916a45dbe5e1fa7a77fda3a64bf Mon Sep 17 00:00:00 2001
From: ChristianWirthContinental
 <117918904+ChristianWirthContinental@users.noreply.github.com>
Date: Tue, 13 Aug 2024 19:04:25 +0200
Subject: [PATCH] Typeguard 4.3.0 (#97)

* typeguard 4.3.0

* Remove fixed version of typeguard

* Lock typeguard version only in test requires

---------

Co-authored-by: Geoff Pleiss <824157+gpleiss@users.noreply.github.com>
---
 setup.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 48daeb1a..f3313a47 100644
--- a/setup.py
+++ b/setup.py
@@ -40,7 +40,6 @@
 install_requires += [
     "scipy",
     "jaxtyping==0.2.19",
-    "typeguard~=2.13.3",
     "mpmath>=0.19,<=1.3",  # avoid incompatibiltiy with torch+sympy with mpmath 1.4
 ]
 
@@ -91,7 +90,13 @@ def find_version(*file_paths):
             "sphinx-autodoc-typehints",
             "uncompyle6<=3.9.0",
         ],
-        "test": ["flake8==5.0.4", "flake8-print==5.0.0", "pytest"],
+        "test": [
+            "flake8==5.0.4",
+            "flake8-print==5.0.0",
+            "pytest",
+            "typeguard~=2.13.3"  # jaxtyping seems to only be compatible with older typeguard versions
+            # https://github.com/patrick-kidger/jaxtyping/commit/77c263c3def8ea3bcb7d7642c5a8402c16cf76fb
+        ],
     },
     test_suite="test",
 )