From 703b68b60baf14e44c365e15d05d1951bb91293d Mon Sep 17 00:00:00 2001
From: Marcel Arpogaus <38564291+MArpogaus@users.noreply.github.com>
Date: Tue, 6 Feb 2024 15:45:15 +0100
Subject: [PATCH] enables extrapolation per default

---
 src/bernstein_flow/bijectors/bernstein.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/bernstein_flow/bijectors/bernstein.py b/src/bernstein_flow/bijectors/bernstein.py
index 903c383..4c2906e 100644
--- a/src/bernstein_flow/bijectors/bernstein.py
+++ b/src/bernstein_flow/bijectors/bernstein.py
@@ -60,7 +60,7 @@ class BernsteinBijector(tfb.AutoCompositeTensorBijector):
     def __init__(
         self,
         thetas: tf.Tensor,
-        extrapolation: str = False,
+        extrapolation: str = True,
         name: str = "bernstein_bijector",
         **kwds,
     ) -> None: