'''simple docstring'''






def    _SCREAMING_SNAKE_CASE		(			__snake_case       : list[int | list]      ,		__snake_case       : int							):
     _A								=     0
     for ele in arr:
          total_sum += product_sum(__snake_case      ,		depth + 1							) if isinstance(__snake_case      ,		__snake_case							) else ele
     return total_sum * depth





def    _SCREAMING_SNAKE_CASE		(			__snake_case       : list[int | list]							):
     return product_sum(__snake_case      ,		1							)


if __name__ == "__main__":
   import doctest

   doctest.testmod()




'''simple docstring'''






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						:    Union[str, 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

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



