diff --git a/GradS/fsrl/policy/lagrangian_base.py b/GradS/fsrl/policy/lagrangian_base.py index 1ab52d2..e45cd93 100644 --- a/GradS/fsrl/policy/lagrangian_base.py +++ b/GradS/fsrl/policy/lagrangian_base.py @@ -250,7 +250,7 @@ def safety_loss(self, values: List, required_GS = True) -> Tuple[torch.tensor, d for index_j in index_list: sim_ij = torch.mean(cos(safety_loss_grad_list[index_i], safety_loss_grad_list[index_j])).item() - sim_list.append(sim_ij) + sim_list.append(sim_ij) if np.max(np.array(sim_list)) < self.sim_threshold and np.min(np.array(sim_list)) > -1 * self.conflicting_threshold: index_list.append(index_i)