

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available, is_vision_available


_UpperCAmelCase		:       int						    =	{
    """configuration_maskformer""": ["""MASKFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP""", """MaskFormerConfig"""],
    """configuration_maskformer_swin""": ["""MaskFormerSwinConfig"""],
}

try:
					if not is_vision_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					_UpperCAmelCase		:       str						    =	["""MaskFormerFeatureExtractor"""]
					_UpperCAmelCase		:       List[str]						    =	["""MaskFormerImageProcessor"""]


try:
					if not is_torch_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					_UpperCAmelCase		:       Union[str, Any]						    =	[
					    """MASKFORMER_PRETRAINED_MODEL_ARCHIVE_LIST""",
					    """MaskFormerForInstanceSegmentation""",
					    """MaskFormerModel""",
					    """MaskFormerPreTrainedModel""",
					]
					_UpperCAmelCase		:       Any						    =	[
					    """MaskFormerSwinBackbone""",
					    """MaskFormerSwinModel""",
					    """MaskFormerSwinPreTrainedModel""",
					]

if TYPE_CHECKING:
					from .configuration_maskformer import MASKFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, MaskFormerConfig
					from .configuration_maskformer_swin import MaskFormerSwinConfig

					try:
										if not is_vision_available():
															raise OptionalDependencyNotAvailable()
					except OptionalDependencyNotAvailable:
										pass
					else:
										from .feature_extraction_maskformer import MaskFormerFeatureExtractor
										from .image_processing_maskformer import MaskFormerImageProcessor
					try:
										if not is_torch_available():
															raise OptionalDependencyNotAvailable()
					except OptionalDependencyNotAvailable:
										pass
					else:
										from .modeling_maskformer import (
										    MASKFORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
										    MaskFormerForInstanceSegmentation,
										    MaskFormerModel,
										    MaskFormerPreTrainedModel,
										)
										from .modeling_maskformer_swin import (
										    MaskFormerSwinBackbone,
										    MaskFormerSwinModel,
										    MaskFormerSwinPreTrainedModel,
										)


else:
					import sys

					_UpperCAmelCase		:       List[Any]						    =	_LazyModule(__name__, globals()["""__file__"""], _import_structure)







import unittest

from transformers import is_torch_available, is_vision_available
from transformers.testing_utils import require_torch, require_vision, slow, torch_device


if is_torch_available():
					import torch

					from transformers import AutoModelForImageClassification

if is_vision_available():
					from transformers import AutoImageProcessor







@require_torch
@require_vision
class 			lowercase    (	unittest.TestCase      ):




				@slow
				def    a							( self	):
											snake_case_          =   AutoImageProcessor.from_pretrained('microsoft/dit-base-finetuned-rvlcdip'	)
											snake_case_          =   AutoModelForImageClassification.from_pretrained('microsoft/dit-base-finetuned-rvlcdip'	)
											model.to(snake_case	)

											from datasets import load_dataset

											snake_case_          =   load_dataset('nielsr/rvlcdip-demo'	)

											snake_case_          =   dataset['train'][0]['image'].convert('RGB'	)

											snake_case_          =   image_processor(snake_case				,       return_tensors='pt'	).to(snake_case	)

											# forward pass
											with torch.no_grad():
																		snake_case_          =   model(**snake_case	)
																		snake_case_          =   outputs.logits

											snake_case_          =   torch.Size((1, 16)	)
											self.assertEqual(logits.shape				,       snake_case	)

											snake_case_          =   torch.tensor(
											    [-0.41_58, -0.40_92, -0.43_47]				,       device=snake_case				,       dtype=torch.float				,       )
											self.assertTrue(torch.allclose(logits[0, :3]				,       snake_case				,       atol=1e-4	)	)




