

import unittest

from transformers import load_tool

from .test_tools_common import ToolTesterMixin

class  __snake_case      (   unittest.TestCase			,SCREAMING_SNAKE_CASE):







							'''simple docstring'''






							def _a      (							self		):
											a__				      =		load_tool("""text-classification"""		)
											self.tool.setup()
											a__				      =		load_tool("""text-classification"""  ,  remote=lowercase_		)


							def _a      (							self		):
											a__				      =		self.tool("""That's quite cool"""  ,  ["""positive""", """negative"""]		)
											self.assertEqual(lowercase_  ,  """positive"""		)


							def _a      (							self		):
											a__				      =		self.remote_tool("""That's quite cool"""  ,  ["""positive""", """negative"""]		)
											self.assertEqual(lowercase_  ,  """positive"""		)


							def _a      (							self		):
											a__				      =		self.tool(text="""That's quite cool"""  ,  labels=["""positive""", """negative"""]		)
											self.assertEqual(lowercase_  ,  """positive"""		)







							def _a      (							self		):
											a__				      =		self.remote_tool(text="""That's quite cool"""  ,  labels=["""positive""", """negative"""]		)
											self.assertEqual(lowercase_  ,  """positive"""		)






from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_torch_available,
)


UpperCAmelCase							 =					{
    """configuration_falcon""": ["""FALCON_PRETRAINED_CONFIG_ARCHIVE_MAP""", """FalconConfig"""],
}

try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	UpperCAmelCase							 =					[
	    """FALCON_PRETRAINED_MODEL_ARCHIVE_LIST""",
	    """FalconForCausalLM""",
	    """FalconModel""",
	    """FalconPreTrainedModel""",
	    """FalconForSequenceClassification""",
	    """FalconForTokenClassification""",
	    """FalconForQuestionAnswering""",
	]


if TYPE_CHECKING:
	from .configuration_falcon import FALCON_PRETRAINED_CONFIG_ARCHIVE_MAP, FalconConfig

	try:
		if not is_torch_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_falcon import (
		    FALCON_PRETRAINED_MODEL_ARCHIVE_LIST,
		    FalconForCausalLM,
		    FalconForQuestionAnswering,
		    FalconForSequenceClassification,
		    FalconForTokenClassification,
		    FalconModel,
		    FalconPreTrainedModel,
		)


else:
	import sys

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

