Merge pull request #180 from thomasckng/patch-1

Update optax requirement and remove testing print statement
kazewong · Aug 22, 2024 · 8125a9a · 8125a9a
2 parents fcb1409 + 8403373
commit 8125a9a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -18,7 +18,7 @@ install_requires =
     jax>=0.4.24
     jaxlib>=0.4.24
     equinox>=0.10.6
-    optax>=0.1.5
+    optax>=0.2.3
     evosax>=0.1.4
     tqdm
     corner

diff --git a/src/flowMC/strategy/global_tuning.py b/src/flowMC/strategy/global_tuning.py
@@ -220,7 +220,6 @@ def __init__(
         **kwargs,
     ):
         class_keys = list(self.__class__.__annotations__.keys())
-        print(class_keys)
         for key, value in kwargs.items():
             if key in class_keys:
                 if not key.startswith("__"):