Removed useless assert

osigaud · Mar 22, 2024 · 40fe046 · 40fe046
1 parent 43edaf3
commit 40fe046
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/bbrl/agents/gymnasium.py b/src/bbrl/agents/gymnasium.py
@@ -172,10 +172,6 @@ def __init__(
         """
         super().__init__(*args, **kwargs)
 
-        assert (
-            include_last_state or reward_at_t
-        ), "Last state should be included when reward is at t+1"
-
         self.reward_at_t = reward_at_t
         self.include_last_state = include_last_state
         self.ghost_params: nn.Parameter = nn.Parameter(torch.Tensor())