Files
Gymnasium/tests/wrappers/utils.py
pseudo-rnd-thoughts 640c509109 Rename to gymnasium
2022-09-08 10:10:07 +01:00

10 lines
275 B
Python

import gymnasium
def has_wrapper(wrapped_env: gymnasium.Env, wrapper_type: type) -> bool:
while isinstance(wrapped_env, gymnasium.Wrapper):
if isinstance(wrapped_env, wrapper_type):
return True
wrapped_env = wrapped_env.env
return False