






"""simple docstring"""







from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


__a							:						int   							=			{
    'configuration_decision_transformer': [
        'DECISION_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP',
        'DecisionTransformerConfig',
    ],
}

try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	__a							:						Dict   							=			[
	    'DECISION_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST',
	    'DecisionTransformerGPT2Model',
	    'DecisionTransformerGPT2PreTrainedModel',
	    'DecisionTransformerModel',
	    'DecisionTransformerPreTrainedModel',
	]


if TYPE_CHECKING:
	from .configuration_decision_transformer import (
	    DECISION_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP,
	    DecisionTransformerConfig,
	)

	try:
		if not is_torch_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_decision_transformer import (
		    DECISION_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
		    DecisionTransformerGPTaModel,
		    DecisionTransformerGPTaPreTrainedModel,
		    DecisionTransformerModel,
		    DecisionTransformerPreTrainedModel,
		)


else:
	import sys

	__a							:						Optional[int]   							=			_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)



def SCREAMING_SNAKE_CASE					(  lowercase_:						int					):
       return n * (2 * n - 1)


def SCREAMING_SNAKE_CASE					(  lowercase_:						int					):
       _UpperCamelCase  :							str	    =	(1 + 24 * n) ** 0.5
       return ((1 + root) / 6) % 1 == 0


def SCREAMING_SNAKE_CASE					(  lowercase_:						int = 144					):
       _UpperCamelCase  :							Optional[Any]	    =	start
       _UpperCamelCase  :							List[str]	    =	hexagonal_num(lowercase_					)
       while not is_pentagonal(lowercase_					):
              n += 1
              _UpperCamelCase  :							Optional[int]	    =	hexagonal_num(lowercase_					)
       return num


if __name__ == "__main__":
 print(f'''{solution()} = ''')



