Add ActNorm tests

davidnabergoj · Aug 24, 2024 · 6fc9e71 · 6fc9e71
1 parent f65fd17
commit 6fc9e71
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/test_reconstruction_bijections.py b/test/test_reconstruction_bijections.py
@@ -11,7 +11,7 @@
 from torchflows.bijections.finite.autoregressive.architectures import NICE, RealNVP, CouplingRQNSF, MAF, IAF, \
     InverseAutoregressiveRQNSF, MaskedAutoregressiveRQNSF
 from torchflows.bijections.finite.autoregressive.layers import ElementwiseScale, ElementwiseAffine, ElementwiseShift, \
-    LRSCoupling, LinearRQSCoupling
+    LRSCoupling, LinearRQSCoupling, ActNorm
 from torchflows.bijections.finite.linear import LU, ReversePermutation, LowerTriangular, Orthogonal, QR
 from torchflows.bijections.finite.residual.architectures import ResFlow, InvertibleResNet, ProximalResFlow
 from torchflows.bijections.finite.residual.iterative import InvertibleResNetBlock, ResFlowBlock
@@ -129,7 +129,8 @@ def assert_valid_reconstruction_continuous(bijection: ContinuousBijection,
     Orthogonal,
     QR,
     ElementwiseAffine,
-    ElementwiseShift
+    ElementwiseShift,
+    ActNorm
 ])
 @pytest.mark.parametrize('batch_shape', __test_constants['batch_shape'])
 @pytest.mark.parametrize('event_shape', __test_constants['event_shape'])