code: def _default_config():
      """Base config to use."""
      return {
          "model": {
              "max_seq_len": 4,
          },
          "embed_dim": 32,
          "num_layers": 2,
          "horizon": 10,
          "num_heads": 2,
          "embed_pdrop": 0.1,
          "resid_pdrop": 0.1,
          "attn_pdrop": 0.1,
          "framework": "torch",
          "lr": 1e-3,
          "lr_schedule": None,
          "optimizer": {
              "weight_decay": 1e-4,
              "betas": [0.9, 0.99],
          },
          "target_return": 200.0,
          "loss_coef_actions": 1.0,
          "loss_coef_obs": 0,
          "loss_coef_returns_to_go": 0,
          "num_gpus": 0,
          "_fake_gpus": None,
          "_enable_new_api_stack": False,
      }
  
file_path: ray/rllib_contrib/dt/tests/test_dt_policy.py
imports: import unittest
  from typing import Dict
  import gymnasium
  import numpy
  from rllib_dt.dt.dt_torch_policy import DTTorchPolicy
  import ray
  from ray.rllib.policy.sample_batch import SampleBatch
  from ray.rllib.utils.framework import try_import_tf, try_import_torch
label_desc: Use image node_id 1 for calling a global function with example usage: _default_config() and returns: dict