Skip to content

Commit df975b2

Browse files
committed
Remove FloatReward. Fixes #794
1 parent 629ef9a commit df975b2

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

tests/algorithms/conftest.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -113,20 +113,10 @@ def pendulum_single_venv(rng) -> VecEnv:
113113
)
114114

115115

116-
# TODO(GH#794): Remove after https://github.com/DLR-RM/stable-baselines3/pull/1676
117-
# merged and released.
118-
class FloatReward(gym.RewardWrapper):
119-
"""Typecasts reward to a float."""
120-
121-
def reward(self, reward):
122-
return float(reward)
123-
124-
125116
@pytest.fixture
126117
def multi_obs_venv() -> VecEnv:
127118
def make_env():
128119
env = envs.SimpleMultiObsEnv(channel_last=False)
129-
env = FloatReward(env)
130120
return RolloutInfoWrapper(env)
131121

132122
return DummyVecEnv([make_env, make_env])

0 commit comments

Comments
 (0)