docstr: Args:
      obs (torch.Tensor): (n_agents, obs_shape)
  Returns:
      self.agent_model(obs, hidden_state)
docstr_tok: ["Args",":","obs","(","torch.Tensor",")",":","(","n_agents",",","obs_shape",")","Returns",":","self.agent_model","(","obs",",","hidden_state",")"]
code: def predict_local_q(self, obs, hidden_state):
      """
      Args:
          obs (torch.Tensor): (n_agents, obs_shape)
      Returns:
          self.agent_model(obs, hidden_state)
      """
      return self.agent_model(obs, hidden_state)
  
file_path: PARL/parl/algorithms/torch/qmix.py
imports: import torch
  import torch.optim
  import torch.nn.functional
  import parl
  from parl.utils.utils import check_model_method
  import numpy
  from copy import deepcopy
label_desc: Use image node_id 3 for calling the QMIX obj's underlying member method code with example usage: obj.predict_local_q(obs, hidden_state) and returns: self