class_docstr: Informations of each trial from one model inference
class_docstr_tok: ["Informations","of","each","trial","from","one","model","inference"]
docstr: After the trial is finished, reward and return of this trial is updated
docstr_tok: ["After","the","trial","is","finished",",","reward","and","return","of","this","trial","is","updated"]
code: def update_rewards(self, rewards, returns):
      """
      After the trial is finished, reward and return of this trial is updated
      """
      self.rewards = rewards
      self.returns = returns
  
imports: import copy
  import logging
  import numpy
  from gym import spaces
  from schema import Schema, Optional
  import nni
  from nni import ClassArgsValidator
  from nni.tuner import Tuner
  from nni.utils import OptimizeMode, extract_scalar_reward
  from .model import Model
  from .util import set_global_seeds
  from .policy import build_lstm_policy
label_desc: Use image node_id 3 for calling the TrialsInfo obj's underlying member method code with example usage: obj.update_rewards(rewards, returns) without return types