TypeError: RandomNumberGenerator._generator_ctor() takes from 0 to 1 positional arguments but 2 were given

[dmitryponv]

Currently throwing error in Training_torch collab notebook
on Line train(progress_fn=progress)

"TypeError: RandomNumberGenerator._generator_ctor() takes from 0 to 1 positional arguments but 2 were given"

---

TypeError Traceback (most recent call last)
in <cell line: 24>()
22 plt.show()
23
---> 24 train(progress_fn=progress)
25
26 print(f'time to jit: {times[1] - times[0]}')

5 frames
in train(env_name, num_envs, episode_length, device, num_timesteps, eval_frequency, unroll_length, batch_size, num_minibatches, num_update_epochs, reward_scaling, entropy_cost, discounting, learning_rate, progress_fn)
67 episode_length=episode_length,
68 backend='spring')
---> 69 env = gym_wrapper.VectorGymWrapper(env)
70 # automatically convert between jax ndarrays and torch tensors:
71 env = torch_wrapper.TorchWrapper(env, device=device)

/usr/local/lib/python3.10/dist-packages/brax/envs/wrappers/gym.py in init(self, env, seed, backend)
114 obs = np.inf * np.ones(self._env.observation_size, dtype='float32')
115 obs_space = spaces.Box(-obs, obs, dtype='float32')
--> 116 self.observation_space = utils.batch_space(obs_space, self.num_envs)
117
118 action = jax.tree.map(np.array, self._env.sys.actuator.ctrl_range)

/usr/lib/python3.10/functools.py in wrapper(*args, **kw)
887 '1 positional argument')
888
--> 889 return dispatch(args[0].class)(*args, **kw)
890
891 funcname = getattr(func, 'name', 'singledispatch function')

/usr/local/lib/python3.10/dist-packages/gym/vector/utils/spaces.py in _batch_space_box(space, n)
48 repeats = tuple([n] + [1] * space.low.ndim)
49 low, high = np.tile(space.low, repeats), np.tile(space.high, repeats)
---> 50 return Box(low=low, high=high, dtype=space.dtype, seed=deepcopy(space.np_random))
51
52

/usr/lib/python3.10/copy.py in deepcopy(x, memo, _nil)
170 y = x
171 else:
--> 172 y = _reconstruct(x, memo, *rv)
173
174 # If is its own copy, don't memoize.

/usr/lib/python3.10/copy.py in _reconstruct(x, memo, func, args, state, listiter, dictiter, deepcopy)
263 if deep and args:
264 args = (deepcopy(arg, memo) for arg in args)
--> 265 y = func(*args)
266 if deep:
267 memo[id(x)] = y

TypeError: RandomNumberGenerator._generator_ctor() takes from 0 to 1 positional arguments but 2 were given