ensure numpy

Author: Xmaster6y

test/test_loggers.py

@@ -488,7 +488,7 @@ def test_log_video(self, trackio_logger):
         # C - number of image channels (e.g. 3 for RGB), H, W - image dimensions.
         # the first 64 frames are black and the next 64 are white
         video = torch.cat(
-            (torch.zeros(128, 1, 32, 32), torch.full((128, 1, 32, 32), 255))
+            (torch.zeros(128, 3, 32, 32), torch.full((128, 3, 32, 32), 255))
         )
         video = video[None, :]
         trackio_logger.log_video(

torchrl/record/loggers/trackio.py

@@ -8,6 +8,8 @@
 
 from collections.abc import Sequence
 
+import numpy as np
+
 from torch import Tensor
 
 from .common import Logger
@@ -96,7 +98,11 @@ def log_video(self, name: str, video: Tensor, **kwargs) -> None:
         fps = kwargs.pop("fps", self.video_fps)
         format = kwargs.pop("format", "mp4")
         self.experiment.log(
-            {name: trackio.Video(video, fps=fps, format=format)},
+            {
+                name: trackio.Video(
+                    video.numpy().astype(np.uint8), fps=fps, format=format
+                )
+            },
             **kwargs,
         )