increases order of f1 bijectors

MArpogaus · Sep 15, 2023 · c1a5c3f · c1a5c3f · github-actions · Sep 15, 2023
1 parent bee520a
commit c1a5c3f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/bernstein_flow/distributions/bernstein_flow.py b/src/bernstein_flow/distributions/bernstein_flow.py
@@ -123,12 +123,12 @@ def init_bijectors(
  bijectors = []
 
  # f1: ŷ = sigma(a1(x)*y - b1(x))
- if tf.is_tensor(a1):
- f1_scale = tfb.Scale(a1, name="scale1")
- bijectors.append(f1_scale)
  if tf.is_tensor(b1):
  f1_shift = tfb.Shift(b1, name="shift1")
  bijectors.append(f1_shift)
+ if tf.is_tensor(a1):
+ f1_scale = tfb.Scale(a1, name="scale1")
+ bijectors.append(f1_scale)
 
  # clip to domain [0, 1]
  if clip_to_bernstein_domain: