




"""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   UpperCAmelCase_    (			unittest.TestCase		):
		@slow
		def 	_lowerCamelCase							(						self       )						->       List[Any]:
					__lowercase			:  int								=       TFXLMRobertaModel.from_pretrained('''jplu/tf-xlm-roberta-base'''       )

					__lowercase			:  str								=       {
					    '''input_ids''': tf.convert_to_tensor([[0, 26_46, 1_02_69, 83, 9_99_42, 2]]				,						dtype=tf.intaa       ),  # "My dog is cute"
					    '''attention_mask''': tf.convert_to_tensor([[1, 1, 1, 1, 1, 1]]				,						dtype=tf.intaa       ),
					}

					__lowercase			:  Union[str, Any]								=       model(UpperCamelCase_       )['''last_hidden_state''']
					__lowercase			:  Optional[int]								=       tf.TensorShape((1, 6, 7_68)       )
					self.assertEqual(output.shape				,						UpperCamelCase_       )
					# compare the actual values for a slice.
					__lowercase			:  Tuple								=       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'''
import pytest
import requests

from datasets.utils.file_utils import http_head

from .utils import OfflineSimulationMode, RequestWouldHangIndefinitelyError, offline







@pytest.mark.integration
def       a_				(	):
							with offline(OfflineSimulationMode.CONNECTION_TIMES_OUT ):
														with pytest.raises(UpperCamelCase_ ):
																					requests.request("GET"						,  "https://huggingface.co" )
														with pytest.raises(requests.exceptions.ConnectTimeout ):
																					requests.request("GET"						,  "https://huggingface.co"						,  timeout=1.0 )







@pytest.mark.integration
def       a_				(	):
							with offline(OfflineSimulationMode.CONNECTION_FAILS ):
														with pytest.raises(requests.exceptions.ConnectionError ):
																					requests.request("GET"						,  "https://huggingface.co" )







def       a_				(	):
							with offline(OfflineSimulationMode.HF_DATASETS_OFFLINE_SET_TO_1 ):
														with pytest.raises(UpperCamelCase_ ):
																					http_head("https://huggingface.co" )





