From 1180ad9e88a911d17f70883061e1cc4030da9d12 Mon Sep 17 00:00:00 2001 From: Simon Dirmeier Date: Sun, 18 Aug 2024 19:48:33 +0200 Subject: [PATCH] Add examples as a run CI --- surjectors/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/surjectors/__init__.py b/surjectors/__init__.py index bb945c5..1745dc8 100644 --- 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", ]