leave bnaf bijection to throw error when sampling

flowjax/bijections/abc.py

@@ -1,6 +1,7 @@
 from abc import ABC, abstractmethod
 import jax.numpy as jnp
 
+
 class Bijection(ABC):
     """Basic bijection class. All bijections should support conditioning variables
     (even if ignored)."""

flowjax/flows.py

@@ -190,9 +190,3 @@ def __init__(
         )
         bijection = Chain(bijections)
         super().__init__(bijection, target_dim, base_log_prob, base_sample)
-
-    def sample(self, *args, **kwargs):
-        raise NotImplementedError(
-            "Sampling this flow would require numerical inversion of the bijection"
-        )
-