diff --git a/gym-unity/gym_unity/envs/unity_env.py b/gym-unity/gym_unity/envs/unity_env.py index cb24d6e51a..fb824b96b1 100755 --- a/gym-unity/gym_unity/envs/unity_env.py +++ b/gym-unity/gym_unity/envs/unity_env.py @@ -44,7 +44,7 @@ def __init__(self, environment_filename: str, worker_id=0, use_visual=False, mul if use_visual and brain.number_visual_observations == 0: raise UnityGymException("`use_visual` was set to True, however there are no" " visual observations as part of this environment.") - self.use_visual = brain.number_visual_observations == 1 and use_visual + self.use_visual = brain.number_visual_observations >= 1 and use_visual if brain.num_stacked_vector_observations != 1: raise UnityGymException(