From e2b05db54680804e4f7aa1559e84df6d6c67ef4f Mon Sep 17 00:00:00 2001 From: dmarzoug Date: Mon, 25 Mar 2024 10:31:12 +0100 Subject: [PATCH] lint --- moojoco/environment/mjc_env.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/moojoco/environment/mjc_env.py b/moojoco/environment/mjc_env.py index a710ab0..7c2a889 100644 --- a/moojoco/environment/mjc_env.py +++ b/moojoco/environment/mjc_env.py @@ -277,7 +277,11 @@ def reset( self, rng: List[np.random.RandomState], *args, **kwargs ) -> VectorMJCEnvState: self._states = list( - self._pool.map(lambda env, sub_rng: env.reset(sub_rng, *args, **kwargs), self._envs, rng) + self._pool.map( + lambda env, sub_rng: env.reset(sub_rng, *args, **kwargs), + self._envs, + rng, + ) ) return self._merged_states