Add orbax-checkpoint as dependancy

tttc3 · tttc3 · commit 4fedf8e690be · 2023-09-25T19:54:33.000+01:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -37,6 +37,7 @@ classifiers = [
 dependencies = [
     "matplotlib",
     "numpy",
+    "orbax-checkpoint",
     "scikit-learn",
     "scikit-optimize>=0.9.0",
     "scipy",
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,6 @@
 matplotlib
 numpy
+orbax-checkpoint
 scikit-learn
 scikit-optimize>=0.9.0
 scipy
