Add examples as a run CI

dirmeier · Aug 18, 2024 · 1180ad9 · 1180ad9
1 parent 376e59a
commit 1180ad9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/surjectors/__init__.py b/surjectors/__init__.py
@@ -1,8 +1,8 @@
 """surjectors: Surjection layers for density estimation with normalizing flows."""
 
-__version__ = "0.3.3"
+__version__ = "0.3.4"
 
-from distrax import ScalarAffine
+from distrax import RationalQuadraticSpline, ScalarAffine
 
 from surjectors._src.bijectors.affine_masked_autoregressive import (
     AffineMaskedAutoregressive,
@@ -63,4 +63,5 @@
     "Slice",
     # "Augment",
     "ScalarAffine",
+    "RationalQuadraticSpline",
 ]