beartype install

homerjed · Sep 29, 2024 · 7881098 · 7881098
1 parent 40268eb
commit 7881098
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "sbgm"
-version = "0.0.15"
+version = "0.0.16"
 description = "Score-based Diffusion models in JAX."
 readme = "README.md"
 requires-python ="~=3.12"
@@ -30,6 +30,7 @@ dependencies = [
     'equinox>=0.11.5',
     'diffrax>=0.6.0',
     'optax>=0.2.3',
+    'beartype',
     'ml_collections',
     'numpy',
     'matplotlib',