Fix test

Author: Louis Faury

torchrl/data/tensor_specs.py

@@ -5750,7 +5750,7 @@ def rand(self, shape: torch.Size = None) -> TensorDictBase:
             _dict,
             batch_size=_size([*shape, *_remove_neg_shapes(self.shape)]),
             device=self.device,
-            names=self.names,
+            names=self._td_dim_names,
         )
 
     def keys(
@@ -6031,7 +6031,7 @@ def zero(self, shape: torch.Size = None) -> TensorDictBase:
             },
             batch_size=_size([*shape, *self._safe_shape]),
             device=device,
-            names=self.names,
+            names=self._td_dim_names,
         )
 
     def __eq__(self, other: object) -> bool: