
'''simple docstring'''







from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


__SCREAMING_SNAKE_CASE		:  Any        =      {
    'configuration_decision_transformer': [
        'DECISION_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP',
        'DecisionTransformerConfig',
    ],
}

try:
					if not is_torch_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					__SCREAMING_SNAKE_CASE		:  Optional[Any]        =      [
					    '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

					__SCREAMING_SNAKE_CASE		:  List[str]        =      _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)

'''simple docstring'''







def 				_snake_case (		lowercase						)       ->       int:
				if not isinstance(lowercase			,    lowercase						):
								__a       :     Union[str, Any]								=	F"""Input value of [number={number}] must be an integer"""
								raise TypeError(lowercase						)
				if number < 1:
								raise ValueError("""Input must be a positive integer"""						)
				return number * (2 * number - 1)


if __name__ == "__main__":
					import doctest

					doctest.testmod()