


from __future__ import annotations




def  lowerCAmelCase_ (							UpperCamelCase_  )     ->		list[int]:
       if len(UpperCamelCase_  ) == 0:
              return array

       UpperCamelCase_       , UpperCamelCase_          =  min(UpperCamelCase_  ), max(UpperCamelCase_  )

       # Compute the variables
       UpperCamelCase_          =  _max - _min + 1
       UpperCamelCase_       , UpperCamelCase_          =  [0] * holes_range, [0] * holes_range

       # Make the sorting.
       for i in array:
              UpperCamelCase_          =  i - _min
              UpperCamelCase_          =  i
              holes_repeat[index] += 1

       # Makes the array back by replacing the numbers.
       UpperCamelCase_          =  0
       for i in range(UpperCamelCase_  ):
              while holes_repeat[i] > 0:
                     UpperCamelCase_          =  holes[i]
                     index += 1
                     holes_repeat[i] -= 1

    # Returns the sorted array.
       return array


if __name__ == "__main__":
     import doctest

     doctest.testmod()
     _UpperCAmelCase        =				input('Enter numbers separated by comma:\n')
     _UpperCAmelCase        =				[int(x) for x in user_input.split(',')]
     print(pigeon_sort(unsorted))








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

