Remove FloatReward. Fixes #794

HumanCompatibleAI · Dec 11, 2023 · df975b2 · df975b2
1 parent 629ef9a
commit df975b2
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/tests/algorithms/conftest.py b/tests/algorithms/conftest.py
@@ -113,20 +113,10 @@ def pendulum_single_venv(rng) -> VecEnv:
     )
 
 
-# TODO(GH#794): Remove after https://github.com/DLR-RM/stable-baselines3/pull/1676
-# merged and released.
-class FloatReward(gym.RewardWrapper):
-    """Typecasts reward to a float."""
-
-    def reward(self, reward):
-        return float(reward)
-
-
 @pytest.fixture
 def multi_obs_venv() -> VecEnv:
     def make_env():
         env = envs.SimpleMultiObsEnv(channel_last=False)
-        env = FloatReward(env)
         return RolloutInfoWrapper(env)
 
     return DummyVecEnv([make_env, make_env])