'''simple docstring'''






from __future__ import annotations

import unittest

from transformers import is_tf_available
from transformers.testing_utils import require_sentencepiece, require_tf, require_tokenizers, slow


if is_tf_available():
    import numpy as np
    import tensorflow as tf

    from transformers import TFXLMRobertaModel




@require_tf
@require_sentencepiece
@require_tokenizers
class    a (       unittest.TestCase				):



  """simple docstring"""







  @slow
  def    lowercase_  (							self					):

    '''simple docstring'''







    __UpperCAmelCase:					List[Any]		= TFXLMRobertaModel.from_pretrained("""jplu/tf-xlm-roberta-base"""					)

    __UpperCAmelCase:					Union[str, Any]		= {
        """input_ids""": tf.convert_to_tensor([[0, 2646, 1_0269, 83, 9_9942, 2]]   ,    dtype=tf.intaa					),  # "My dog is cute"
        """attention_mask""": tf.convert_to_tensor([[1, 1, 1, 1, 1, 1]]   ,    dtype=tf.intaa					),
    }

    __UpperCAmelCase:					List[str]		= model(snake_case_					)["""last_hidden_state"""]
    __UpperCAmelCase:					Optional[Any]		= tf.TensorShape((1, 6, 768)					)
    self.assertEqual(output.shape   ,    snake_case_					)
    # compare the actual values for a slice.
    __UpperCAmelCase:					Dict		= tf.convert_to_tensor(
        [
            [
                [0.0_6_8_1_7_6_2, 0.1_0_8_9_4_4_5_1, 0.0_6_7_7_2_5_0_4],
                [-0.0_6_4_2_3_6_6_8, 0.0_2_3_6_6_6_1_5, 0.0_4_3_2_9_3_4_4],
                [-0.0_6_0_5_7_2_9_5, 0.0_9_9_7_4_1_3_5, -0.0_0_0_7_0_5_8_4],
            ]
        ]   ,    dtype=tf.floataa   ,    )

    self.assertTrue(np.allclose(output[:, :3, :3].numpy()   ,    expected_slice.numpy()   ,    atol=1e-4					)					)
'''simple docstring'''






from __future__ import annotations

SCREAMING_SNAKE_CASE_	   =			10






def  UpperCamelCase__					(					_lowercase    :				list[int]     )							->	list[int]:
  __UpperCAmelCase:					Union[str, Any]		= 1
  __UpperCAmelCase:					Optional[Any]		= max(_lowercase     )
  while placement <= max_digit:
    # declare and initialize empty buckets
    __UpperCAmelCase:					list[list]		= [[] for _ in range(_lowercase     )]
    # split list_of_ints between the buckets
    for i in list_of_ints:
      __UpperCAmelCase:					Optional[Any]		= int((i / placement) % RADIX     )
      buckets[tmp].append(_lowercase     )
    # put each buckets' contents into list_of_ints
    __UpperCAmelCase:					Optional[int]		= 0
    for b in range(_lowercase     ):
      for i in buckets[b]:
        __UpperCAmelCase:					str		= i
        a += 1
        # move to next
    placement *= RADIX
  return list_of_ints


if __name__ == "__main__":
    import doctest

    doctest.testmod()