code: def _dummy_model_config():
      """Model config with dummy poliy nets for two sub-steps."""
      network_config = [
          {
              "_target_": "maze.test.shared_test_utils.dummy_models.actor_model.DummyPolicyNet",
              "non_lin": "torch.nn.SELU",
          }
      ]
      return {
          "_target_": CustomModelComposer,
          "distribution_mapper_config": {},
          "policy": {
              "_target_": "maze.perception.models.policies.ProbabilisticPolicyComposer",
              "networks": network_config * 2,
              "substeps_with_separate_agent_nets": [],
          },
          "critic": None,
      }
  
file_path: maze/maze/test/core/agent/test_multi_agent_torch_policy.py
imports: from typing import Dict, Optional
  from maze.core.env.maze_env import MazeEnv
  from maze.core.env.structured_env import ActorID
  from maze.perception.models.custom_model_composer import CustomModelComposer
  from maze.test.shared_test_utils.helper_functions import build_dummy_structured_env
  from maze.test.shared_test_utils.dummy_models.actor_model import DummyPolicyNet
label_desc: Use image node_id 1 for calling a global function with example usage: _dummy_model_config() and returns: dict