Skip to content

Commit 68cec40

Browse files
committed
Fix gym default step docstring
1 parent 77d188f commit 68cec40

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

stable_baselines3/common/envs/bit_flipping_env.py

+6
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,12 @@ def reset(self, seed: Optional[int] = None) -> Dict[str, Union[int, np.ndarray]]
165165
return self._get_obs()
166166

167167
def step(self, action: Union[np.ndarray, int]) -> GymStepReturn:
168+
"""
169+
Step into the env.
170+
171+
:param action:
172+
:return:
173+
"""
168174
if self.continuous:
169175
self.state[action > 0] = 1 - self.state[action > 0]
170176
else:

0 commit comments

Comments
 (0)