We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 629ef9a commit df975b2Copy full SHA for df975b2
tests/algorithms/conftest.py
@@ -113,20 +113,10 @@ def pendulum_single_venv(rng) -> VecEnv:
113
)
114
115
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
125
@pytest.fixture
126
def multi_obs_venv() -> VecEnv:
127
def make_env():
128
env = envs.SimpleMultiObsEnv(channel_last=False)
129
- env = FloatReward(env)
130
return RolloutInfoWrapper(env)
131
132
return DummyVecEnv([make_env, make_env])
0 commit comments