MultiDiscreteTensorSpec expects an nvec, not an n

pytorch · Jun 6, 2024 · 9c7cc36 · 9c7cc36
1 parent 013d110
commit 9c7cc36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/data/tensor_specs.py b/torchrl/data/tensor_specs.py
@@ -3369,7 +3369,7 @@ def to(self, dest: Union[torch.dtype, DEVICE_TYPING]) -> CompositeSpec:
             return self
         mask = self.mask.to(dest) if self.mask is not None else None
         return self.__class__(
-            n=self.nvec.to(dest),
+            nvec=self.nvec.to(dest),
             shape=None,
             device=dest_device,
             dtype=dest_dtype,