

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


SCREAMING_SNAKE_CASE_						=					{
    '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_						=					[
					    '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_						=					_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)










"""simple docstring"""







import importlib.metadata
import subprocess
import sys





def 						lowercase__(A = False						)				->List[str]:




						"""simple docstring"""



						lowercase__				:      Any=       False
						if "IPython" in sys.modules:
												lowercase__				:      Tuple=       "google.colab" in str(sys.modules["IPython"].get_ipython()						)

						if in_colab:
												if upgrade:
																		lowercase__				:      Optional[int]=       ["pip", "install", "-U", "torch"]
																		subprocess.run(A					,		check=A						)
												# get the current version of torch
												lowercase__				:      Tuple=       importlib.metadata.version("torch"						)
												lowercase__				:      Dict=       torch_version[: torch_version.rindex("."						)]
												lowercase__				:      Dict=       f'''https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-{torch_version_trunc}-cp37-cp37m-linux_x86_64.whl'''
												lowercase__				:      Union[str, Any]=       ["pip", "install", xla_wheel]
												subprocess.run(A					,		check=A						)
						else:
												raise RuntimeError("`install_xla` utility works only on google colab."						)






