code: def on_policy_output(self, ac_data: ActionConnectorDataType):
      # Buffer latest output states for next input __call__.
      self._states[ac_data.env_id][ac_data.agent_id] = ac_data.output
  
imports: from collections import defaultdict
  import logging
  import pickle
  from typing import Any
  import numpy
  from ray.rllib.utils.annotations import override
  import tree
  from ray.rllib.connectors.connector import AgentConnector, Connector, ConnectorContext
  from ray import cloudpickle
  from ray.rllib.connectors.registry import register_connector
  from ray.rllib.policy.sample_batch import SampleBatch
  from ray.rllib.core.models.base import STATE_OUT
  from ray.rllib.utils.spaces.space_utils import get_base_struct_from_space
  from ray.rllib.utils.typing import ActionConnectorDataType, AgentConnectorDataType
  from ray.util.annotations import PublicAPI
label_desc: Use image node_id 4 for calling the StateBufferConnector obj's underlying member method code with example usage: obj.on_policy_output(ac_data) without return types