


'''simple docstring'''



import sys
import tempfile
import unittest
import unittest.mock as mock
from pathlib import Path

from huggingface_hub import HfFolder, delete_repo
from requests.exceptions import HTTPError

from transformers import AutoImageProcessor, ViTImageProcessor
from transformers.testing_utils import TOKEN, USER, get_tests_dir, is_staging_test


sys.path.append(str(Path(__file__).parent.parent / """utils"""))

from test_module.custom_image_processing import CustomImageProcessor  # noqa E402


_lowercase    :    List[str]			     =  get_tests_dir("""fixtures""")






class  lowerCamelCase_(     unittest.TestCase ):




				def   a__(	self    :				Any   )->  str:

										"""simple docstring"""




										UpperCAmelCase         =			mock.Mock()
										UpperCAmelCase         =			500
										UpperCAmelCase         =			{}
										UpperCAmelCase         =			HTTPError
										UpperCAmelCase         =			{}

										# Download this model to make sure it's in the cache.
										UpperCAmelCase         =			ViTImageProcessor.from_pretrained('''hf-internal-testing/tiny-random-vit'''   )
										# Under the mock environment we get a 500 error when trying to reach the model.
										with mock.patch('''requests.Session.request'''   ,					return_value=_SCREAMING_SNAKE_CASE   ) as mock_head:
																UpperCAmelCase         =			ViTImageProcessor.from_pretrained('''hf-internal-testing/tiny-random-vit'''   )
																# This check we did call the fake head request
																mock_head.assert_called()




				def   a__(	self    :				int   )->  Tuple:

										"""simple docstring"""




										UpperCAmelCase         =			ViTImageProcessor.from_pretrained(
										    '''https://huggingface.co/hf-internal-testing/tiny-random-vit/resolve/main/preprocessor_config.json'''   )





				def   a__(	self    :				Optional[int]   )->  Union[str, Any]:

										"""simple docstring"""




										with self.assertRaises(_SCREAMING_SNAKE_CASE   ):
																# config is in subfolder, the following should not work without specifying the subfolder
																UpperCAmelCase         =			AutoImageProcessor.from_pretrained('''hf-internal-testing/stable-diffusion-all-variants'''   )

										UpperCAmelCase         =			AutoImageProcessor.from_pretrained(
										    '''hf-internal-testing/stable-diffusion-all-variants'''   ,					subfolder='''feature_extractor'''   )

										self.assertIsNotNone(_SCREAMING_SNAKE_CASE   )







@is_staging_test
class  lowerCamelCase_(     unittest.TestCase ):




				@classmethod
				def   a__(	cls    :				Any   )->  List[Any]:

										"""simple docstring"""




										UpperCAmelCase         =			TOKEN
										HfFolder.save_token(_SCREAMING_SNAKE_CASE   )




				@classmethod
				def   a__(	cls    :				Optional[Any]   )->  Dict:

										"""simple docstring"""




										try:
																delete_repo(token=cls._token   ,					repo_id='''test-image-processor'''   )
										except HTTPError:
																pass

										try:
																delete_repo(token=cls._token   ,					repo_id='''valid_org/test-image-processor-org'''   )
										except HTTPError:
																pass

										try:
																delete_repo(token=cls._token   ,					repo_id='''test-dynamic-image-processor'''   )
										except HTTPError:
																pass




				def   a__(	self    :				Dict   )->  int:

										"""simple docstring"""




										UpperCAmelCase         =			ViTImageProcessor.from_pretrained(_SCREAMING_SNAKE_CASE   )
										image_processor.push_to_hub('''test-image-processor'''   ,					use_auth_token=self._token   )

										UpperCAmelCase         =			ViTImageProcessor.from_pretrained(F"""{USER}/test-image-processor"""   )
										for k, v in image_processor.__dict__.items():
																self.assertEqual(_SCREAMING_SNAKE_CASE   ,					getattr(_SCREAMING_SNAKE_CASE   ,					_SCREAMING_SNAKE_CASE   )   )

										# Reset repo
										delete_repo(token=self._token   ,					repo_id='''test-image-processor'''   )

										# Push to hub via save_pretrained
										with tempfile.TemporaryDirectory() as tmp_dir:
																image_processor.save_pretrained(
																    _SCREAMING_SNAKE_CASE   ,					repo_id='''test-image-processor'''   ,					push_to_hub=_SCREAMING_SNAKE_CASE   ,					use_auth_token=self._token   )

										UpperCAmelCase         =			ViTImageProcessor.from_pretrained(F"""{USER}/test-image-processor"""   )
										for k, v in image_processor.__dict__.items():
																self.assertEqual(_SCREAMING_SNAKE_CASE   ,					getattr(_SCREAMING_SNAKE_CASE   ,					_SCREAMING_SNAKE_CASE   )   )




				def   a__(	self    :				Union[str, Any]   )->  List[Any]:

										"""simple docstring"""




										UpperCAmelCase         =			ViTImageProcessor.from_pretrained(_SCREAMING_SNAKE_CASE   )
										image_processor.push_to_hub('''valid_org/test-image-processor'''   ,					use_auth_token=self._token   )

										UpperCAmelCase         =			ViTImageProcessor.from_pretrained('''valid_org/test-image-processor'''   )
										for k, v in image_processor.__dict__.items():
																self.assertEqual(_SCREAMING_SNAKE_CASE   ,					getattr(_SCREAMING_SNAKE_CASE   ,					_SCREAMING_SNAKE_CASE   )   )

										# Reset repo
										delete_repo(token=self._token   ,					repo_id='''valid_org/test-image-processor'''   )

										# Push to hub via save_pretrained
										with tempfile.TemporaryDirectory() as tmp_dir:
																image_processor.save_pretrained(
																    _SCREAMING_SNAKE_CASE   ,					repo_id='''valid_org/test-image-processor-org'''   ,					push_to_hub=_SCREAMING_SNAKE_CASE   ,					use_auth_token=self._token   )

										UpperCAmelCase         =			ViTImageProcessor.from_pretrained('''valid_org/test-image-processor-org'''   )
										for k, v in image_processor.__dict__.items():
																self.assertEqual(_SCREAMING_SNAKE_CASE   ,					getattr(_SCREAMING_SNAKE_CASE   ,					_SCREAMING_SNAKE_CASE   )   )





				def   a__(	self    :				Tuple   )->  List[Any]:

										"""simple docstring"""




										CustomImageProcessor.register_for_auto_class()
										UpperCAmelCase         =			CustomImageProcessor.from_pretrained(_SCREAMING_SNAKE_CASE   )

										image_processor.push_to_hub('''test-dynamic-image-processor'''   ,					use_auth_token=self._token   )

										# This has added the proper auto_map field to the config
										self.assertDictEqual(
										    image_processor.auto_map   ,					{'''AutoImageProcessor''': '''custom_image_processing.CustomImageProcessor'''}   ,					)

										UpperCAmelCase         =			AutoImageProcessor.from_pretrained(
										    F"""{USER}/test-dynamic-image-processor"""   ,					trust_remote_code=_SCREAMING_SNAKE_CASE   )
										# Can't make an isinstance check because the new_image_processor is from the CustomImageProcessor class of a dynamic module
										self.assertEqual(new_image_processor.__class__.__name__   ,					'''CustomImageProcessor'''   )









'''simple docstring'''



# Copyright 2022 The HuggingFace Team and The OpenBMB Team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from typing import TYPE_CHECKING

# rely on isort to merge the imports
from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_tokenizers_available, is_torch_available


_lowercase    :    Any			     =  {
    """configuration_cpmant""": ["""CPMANT_PRETRAINED_CONFIG_ARCHIVE_MAP""", """CpmAntConfig"""],
    """tokenization_cpmant""": ["""CpmAntTokenizer"""],
}

try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	_lowercase    :    int			     =  [
	    """CPMANT_PRETRAINED_MODEL_ARCHIVE_LIST""",
	    """CpmAntForCausalLM""",
	    """CpmAntModel""",
	    """CpmAntPreTrainedModel""",
	]


if TYPE_CHECKING:
	from .configuration_cpmant import CPMANT_PRETRAINED_CONFIG_ARCHIVE_MAP, CpmAntConfig
	from .tokenization_cpmant import CpmAntTokenizer

	try:
		if not is_torch_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_cpmant import (
		    CPMANT_PRETRAINED_MODEL_ARCHIVE_LIST,
		    CpmAntForCausalLM,
		    CpmAntModel,
		    CpmAntPreTrainedModel,
		)


else:
	import sys

	_lowercase    :    Dict			     =  _LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)





