Skip to content

Commit

Permalink
fix side effects baseline test flakiness due to corner case
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 463419361
  • Loading branch information
vkrakovna authored and diegolascasas committed Oct 12, 2022
1 parent 6fcb842 commit 586a1c5
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions side_effects_penalties/side_effects_penalty_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,8 @@ def _test_trajectory(self, actions, key):
}
self.assertEqual(baseline_state, comparison_dict[key])
current_state = next_state
if timestep.last():
return


class StartBaselineTest(BaselineTestCase):
Expand Down

0 comments on commit 586a1c5

Please sign in to comment.