Fix bijection inversion call

davidnabergoj · Mar 21, 2024 · 8d6a5a7 · 8d6a5a7
1 parent 3dab74e
commit 8d6a5a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/normalizing_flows/bijections/finite/autoregressive/layers.py b/normalizing_flows/bijections/finite/autoregressive/layers.py
@@ -70,7 +70,7 @@ def __init__(self,
         if event_shape == (1,):
             raise ValueError
         coupling = make_coupling(event_shape, edge_list)
-        transformer = Affine(event_shape=torch.Size((coupling.target_event_size,))).invert()
+        transformer = invert(Affine(event_shape=torch.Size((coupling.target_event_size,))))
         conditioner_transform = FeedForward(
             input_event_shape=torch.Size((coupling.source_event_size,)),
             parameter_shape=torch.Size(transformer.parameter_shape),