From 40fe0468feb8998e62c3cd6bb3a575fef88e256f Mon Sep 17 00:00:00 2001
From: Benjamin Piwowarski <benjamin@piwowarski.fr>
Date: Fri, 22 Mar 2024 15:29:29 +0100
Subject: [PATCH] Removed useless assert

---
 src/bbrl/agents/gymnasium.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/src/bbrl/agents/gymnasium.py b/src/bbrl/agents/gymnasium.py
index ff1e641..1fab54a 100644
--- 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())