Remove inappropriate warning (warns about supported case according to docstring)

This commit is contained in:
Dominik Jain 2024-01-11 17:08:54 +01:00
parent ff398beed9
commit 24b7b82e56

View File

@ -97,7 +97,6 @@ class Collector:
) -> None: ) -> None:
super().__init__() super().__init__()
if isinstance(env, gym.Env) and not hasattr(env, "__len__"): if isinstance(env, gym.Env) and not hasattr(env, "__len__"):
warnings.warn("Single environment detected, wrap to DummyVectorEnv.")
self.env = DummyVectorEnv([lambda: env]) self.env = DummyVectorEnv([lambda: env])
else: else:
self.env = env # type: ignore self.env = env # type: ignore