


'''simple docstring'''






import re


def       _snake_case     (  A )		->					bool:
							lowerCAmelCase__      =			re.compile(
							    R'''^(?:0|94|\+94|0{2}94)''' R'''7(0|1|2|4|5|6|7|8)''' R'''(-| |)''' R'''\d{7}$''' )

							return bool(re.search(A				,					A ) )


if __name__ == "__main__":
		__UpperCAmelCase					       =							'''0094702343221'''

		print(is_sri_lankan_phone_number(phone))



'''simple docstring'''






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       _snake_case     (  A				,					A				,					A )		->					jnp.ndarray:
							lowerCAmelCase__      =			jnp.zeros_like(A )
							lowerCAmelCase__      =			shifted_input_ids.at[:, 1:].set(input_ids[:, :-1] )
							lowerCAmelCase__      =			shifted_input_ids.at[:, 0].set(A )

							lowerCAmelCase__      =			jnp.where(shifted_input_ids == -100				,					A				,					A )
							return shifted_input_ids




class 	a__       (							a__						):





							'''simple docstring'''






							lowercase__    :     int					=   "mt5"
							lowercase__    :     Dict					=   MTaConfig




class 	a__       (							a__						):





							'''simple docstring'''






							lowercase__    :     int					=   "mt5"
							lowercase__    :     Any					=   MTaConfig




class 	a__       (							a__						):





							'''simple docstring'''






							lowercase__    :     Union[str, Any]					=   "mt5"
							lowercase__    :     Tuple					=   MTaConfig