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     UpperCamelCase						(	__lowercase						:			jnp.array						,__lowercase						:			int						,__lowercase						:			int     ):
		'''simple docstring'''




		A_		:    Dict         =				jnp.zeros_like(__lowercase     )
		A_		:    List[str]         =				shifted_input_ids.at[:, 1:].set(input_ids[:, :-1]     )
		A_		:    Optional[Any]         =				shifted_input_ids.at[:, 0].set(__lowercase     )

		A_		:    Union[str, Any]         =				jnp.where(shifted_input_ids == -1_00						,__lowercase						,__lowercase     )
		return shifted_input_ids





class UpperCAmelCase						( __A  ):


				'''simple docstring'''






				lowerCamelCase_    = '''mt5'''
				lowerCamelCase_    = MTaConfig





class UpperCAmelCase						( __A  ):


				'''simple docstring'''






				lowerCamelCase_    = '''mt5'''
				lowerCamelCase_    = MTaConfig





class UpperCAmelCase						( __A  ):


				'''simple docstring'''






				lowerCamelCase_    = '''mt5'''
				lowerCamelCase_    = MTaConfig







import warnings

from ...utils import logging
from .image_processing_donut import DonutImageProcessor


_UpperCAmelCase            =		logging.get_logger(__name__)





class UpperCAmelCase						( __A  ):


				'''simple docstring'''






				def __init__(						self		,     *lowercase		,     **lowercase ):



						"""simple docstring"""





						warnings.warn(
						    'The class DonutFeatureExtractor is deprecated and will be removed in version 5 of Transformers. Please'
						    ' use DonutImageProcessor instead.'		,     lowercase		,     )
						super().__init__(*lowercase		,     **lowercase )






