Traceback (most recent call last):
File "<input>", line 1, in <module>
File "../python3.10/site-packages/torchrl/data/tensor_specs.py", line 3212, in to
return self.__class__(
TypeError: MultiDiscreteTensorSpec.__init__() got an unexpected keyword argument 'n'
Expected behavior
The MultiDiscreteTensorSpec should be sent to device cuda:0.
System info
import torchrl, numpy, sys
print(torchrl.__version__, numpy.__version__, sys.version, sys.platform)
0.4.0 1.26.4 3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0] linux
Describe the bug
When calling
to
on a MultiDiscreteTensorSpec, it fails withTypeError: MultiDiscreteTensorSpec.__init__() got an unexpected keyword argument 'n'
To Reproduce
Steps to reproduce the behavior.
Expected behavior
The MultiDiscreteTensorSpec should be sent to device
cuda:0
.System info
Reason and Possible fixes
Changing to
nvec
here fixes the issue.Checklist