Error in running set_info_dict_reader

[ShahRutav]

While running the code, it leads to an error:

Traceback (most recent call last):
  File "test.py", line 24, in <module>
    env = env.set_info_dict_reader(info_dict_reader=reader)
  File "/home/rutavms/research/robohive/latest/rl/torchrl/envs/gym_like.py", line 284, in set_info_dict_reader
    self.observation_spec[info_key] = spec
  File "/home/rutavms/research/robohive/latest/rl/torchrl/data/tensor_specs.py", line 1257, in __setitem__
    raise RuntimeError(
RuntimeError: Setting a new attribute (solved) on another device (cpu against cuda:0). All devices of CompositeSpec must match.

I locally fixed in torchRL by simply adding spec.to(self.device) here

[vmoens]

closed as of pytorch/rl#824