





def  lowerCAmelCase_ (							UpperCamelCase_  )     ->		int:
       assert (
           isinstance(UpperCamelCase_					,		UpperCamelCase_  ) and number_of_steps > 0
       ), F'''number_of_steps needs to be positive integer, your input {number_of_steps}'''
       if number_of_steps == 1:
              return 1
       UpperCamelCase_       , UpperCamelCase_          =  1, 1
       for _ in range(number_of_steps - 1  ):
              UpperCamelCase_       , UpperCamelCase_          =  current + previous, current
       return current


if __name__ == "__main__":
     import doctest

     doctest.testmod()








import jax.numpy as jnp

from ...utils import logging
from ..ta.modeling_flax_ta import FlaxTaEncoderModel, FlaxTaForConditionalGeneration, FlaxTaModel
from .configuration_mta import MTaConfig


_UpperCAmelCase        =				logging.get_logger(__name__)

_UpperCAmelCase        =				'T5Config'




def  lowerCAmelCase_ (							UpperCamelCase_					,		UpperCamelCase_					,		UpperCamelCase_  )     ->		jnp.ndarray:
       UpperCamelCase_          =  jnp.zeros_like(UpperCamelCase_  )
       UpperCamelCase_          =  shifted_input_ids.at[:, 1:].set(input_ids[:, :-1]  )
       UpperCamelCase_          =  shifted_input_ids.at[:, 0].set(UpperCamelCase_  )

       UpperCamelCase_          =  jnp.where(shifted_input_ids == -100					,		UpperCamelCase_					,		UpperCamelCase_  )
       return shifted_input_ids



class 							_UpperCamelCase					(	lowerCAmelCase_     ):
       _UpperCamelCase				:					Dict         =   '''mt5'''
       _UpperCamelCase				:					Union[str, Any]         =   MTaConfig



class 							_UpperCamelCase					(	lowerCAmelCase_     ):
       _UpperCamelCase				:					Dict         =   '''mt5'''
       _UpperCamelCase				:					Optional[int]         =   MTaConfig



class 							_UpperCamelCase					(	lowerCAmelCase_     ):
       _UpperCamelCase				:					Union[str, Any]         =   '''mt5'''
       _UpperCamelCase				:					List[Any]         =   MTaConfig

