


'''simple docstring'''



from typing import TYPE_CHECKING

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


__lowercase     :						Optional[int]	=     {
    'configuration_graphormer': ['GRAPHORMER_PRETRAINED_CONFIG_ARCHIVE_MAP', 'GraphormerConfig'],
}

try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	__lowercase     :						Any	=     [
	    'GRAPHORMER_PRETRAINED_MODEL_ARCHIVE_LIST',
	    'GraphormerForGraphClassification',
	    'GraphormerModel',
	    'GraphormerPreTrainedModel',
	]


if TYPE_CHECKING:
	from .configuration_graphormer import GRAPHORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, GraphormerConfig

	try:
		if not is_torch_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_graphormer import (
		    GRAPHORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
		    GraphormerForGraphClassification,
		    GraphormerModel,
		    GraphormerPreTrainedModel,
		)


else:
	import sys

	__lowercase     :						Dict	=     _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)







'''simple docstring'''



from .imports import is_rich_available


if is_rich_available():
	from rich.traceback import install

	install(show_locals=False)

else:
	raise ModuleNotFoundError('To use the rich extension, install rich with `pip install rich`')



