(self, trajectory, temperature=1.0)
return self._expert_policy_fn(
        collect_model=self._collect_model,
        policy_distribution=self._policy_dist,
        loop=self.loop,
        trajectory_np=trajectory_np,
        temperature=temperature,
    )
Policy function that runs the expert.
Policy function that runs the expert.
Policy function that runs the expert .
def policy(self, trajectory, temperature=1.0):
    """Policy function that runs the expert."""
    tr_slice = trajectory.suffix(self._max_slice_length)
    trajectory_np = tr_slice.to_np(timestep_to_np=self.task.timestep_to_np)
    return self._expert_policy_fn(
        collect_model=self._collect_model,
        policy_distribution=self._policy_dist,
        loop=self.loop,
        trajectory_np=trajectory_np,
        temperature=temperature,
    )
def policy ( self , trajectory , temperature = 1.0 ) : tr_slice = trajectory . suffix ( self . _max_slice_length ) trajectory_np = tr_slice . to_np ( timestep_to_np = self . task . timestep_to_np ) return self . _expert_policy_fn ( collect_model = self . _collect_model , policy_distribution = self . _policy_dist , loop = self . loop , trajectory_np = trajectory_np , temperature = temperature , )