from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


_SCREAMING_SNAKE_CASE   		=			{"""configuration_nat""": ["""NAT_PRETRAINED_CONFIG_ARCHIVE_MAP""", """NatConfig"""]}


try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     _SCREAMING_SNAKE_CASE   		=			[
         """NAT_PRETRAINED_MODEL_ARCHIVE_LIST""",
         """NatForImageClassification""",
         """NatModel""",
         """NatPreTrainedModel""",
         """NatBackbone""",
     ]

if TYPE_CHECKING:
     from .configuration_nat import NAT_PRETRAINED_CONFIG_ARCHIVE_MAP, NatConfig

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_nat import (
              NAT_PRETRAINED_MODEL_ARCHIVE_LIST,
              NatBackbone,
              NatForImageClassification,
              NatModel,
              NatPreTrainedModel,
          )

else:
     import sys

     _SCREAMING_SNAKE_CASE   		=			_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)

import unittest

from datasets import load_dataset

from transformers import load_tool

from .test_tools_common import ToolTesterMixin
class SCREAMING_SNAKE_CASE_		(			unittest.TestCase     ,    __lowerCAmelCase		):







   def 	lowerCamelCase_  (    self	:	Optional[Any]							):





    """simple docstring"""


    UpperCamelCase      =			load_tool("""document-question-answering"""							)
    self.tool.setup()
    UpperCamelCase      =			load_tool("""document-question-answering"""      ,      remote=lowerCamelCase_							)







   def 	lowerCamelCase_  (    self	:	Optional[Any]							):





    """simple docstring"""


    UpperCamelCase      =			load_dataset("""hf-internal-testing/example-documents"""      ,      split="""test"""							)
    UpperCamelCase      =			dataset[0]["""image"""]

    UpperCamelCase      =			self.tool(lowerCamelCase_      ,      """When is the coffee break?"""							)
    self.assertEqual(lowerCamelCase_      ,      """11-14 to 11:39 a.m."""							)







   def 	lowerCamelCase_  (    self	:	Union[str, Any]							):





    """simple docstring"""


    UpperCamelCase      =			load_dataset("""hf-internal-testing/example-documents"""      ,      split="""test"""							)
    UpperCamelCase      =			dataset[0]["""image"""]

    UpperCamelCase      =			self.remote_tool(lowerCamelCase_      ,      """When is the coffee break?"""							)
    self.assertEqual(lowerCamelCase_      ,      """11-14 to 11:39 a.m."""							)







   def 	lowerCamelCase_  (    self	:	int							):





    """simple docstring"""


    UpperCamelCase      =			load_dataset("""hf-internal-testing/example-documents"""      ,      split="""test"""							)
    UpperCamelCase      =			dataset[0]["""image"""]

    self.tool(document=lowerCamelCase_      ,      question="""When is the coffee break?"""							)







   def 	lowerCamelCase_  (    self	:	Any							):





    """simple docstring"""


    UpperCamelCase      =			load_dataset("""hf-internal-testing/example-documents"""      ,      split="""test"""							)
    UpperCamelCase      =			dataset[0]["""image"""]

    UpperCamelCase      =			self.remote_tool(document=lowerCamelCase_      ,      question="""When is the coffee break?"""							)
    self.assertEqual(lowerCamelCase_      ,      """11-14 to 11:39 a.m."""							)
