src/transformers/generation/flax_logits_process.py/FlaxLogitsWarper/__call__
class FlaxLogitsWarper:    def __call__(self, input_ids: jnp.ndarray, scores: jnp.ndarray) -> jnp.ndarray:
        """Flax method for warping logits."""
        raise NotImplementedError(
            f"{self.__class__} is an abstract class. Only classes inheriting this class can be called."
        )