



from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


UpperCAmelCase__			:   Dict      			=    {
    'configuration_decision_transformer': [
        'DECISION_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP',
        'DecisionTransformerConfig',
    ],
}

try:
					if not is_torch_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					UpperCAmelCase__			:   str      			=    [
					    '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

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





UpperCAmelCase__			:   Optional[int]      			=    [0.4_85, 0.4_56, 0.4_06]
UpperCAmelCase__			:   Union[str, Any]      			=    [0.2_29, 0.2_24, 0.2_25]
UpperCAmelCase__			:   Union[str, Any]      			=    [0.5, 0.5, 0.5]
UpperCAmelCase__			:   Optional[Any]      			=    [0.5, 0.5, 0.5]
UpperCAmelCase__			:   str      			=    [0.48_14_54_66, 0.4_57_82_75, 0.40_82_10_73]
UpperCAmelCase__			:   Dict      			=    [0.26_86_29_54, 0.26_13_02_58, 0.27_57_77_11]
