code: def _mock_observation_spaces_dict():
      return {
          0: spaces.Dict(
              {
                  "observation_inventory": spaces.Box(
                      0,
                      1,
                      shape=(
                          8,
                          16,
                      ),
                  ),
                  "observation_screen": spaces.Box(
                      0, 1, shape=(8, 3, 64, 64)
                  ),
              }
          )
      }
  
file_path: maze/docs/test/policy_and_value_networks/test_complex_models.py
imports: import torch
  import yaml
  from gym import spaces
  import docs.source.policy_and_value_networks.code_snippets
  from docs.source.policy_and_value_networks.code_snippets.custom_complex_critic_net import CustomComplexCriticNet
  from docs.source.policy_and_value_networks.code_snippets.custom_complex_policy_net import CustomComplexPolicyNet
  from maze.core.utils.factory import Factory
  from maze.core.utils.structured_env_utils import flat_structured_space
  from maze.distributions.distribution_mapper import DistributionMapper
  from maze.perception.models.model_composer import BaseModelComposer
  from maze.perception.perception_utils import observation_spaces_to_in_shapes
label_desc: Use image node_id 1 for calling a global function with example usage: _mock_observation_spaces_dict() and returns: dict