init

Author: vmoens

test/test_loggers.py

@@ -35,7 +35,9 @@
 else:
     TORCHVISION_VERSION = version.parse("0.0.1")
 
-_has_mp4 = (_has_tv and TORCHVISION_VERSION < version.parse("0.22")) or _has_torchcodec
+_has_mp4 = (
+    _has_tv and version.parse("0.20") <= TORCHVISION_VERSION < version.parse("0.22")
+) or _has_torchcodec
 
 if _has_tb:
     from tensorboard.backend.event_processing.event_accumulator import EventAccumulator

torchrl/data/replay_buffers/storages.py

@@ -1007,6 +1007,7 @@ def _set_tree_map(self, cursor, data, storage):  # noqa: 534
         for datum, store in zip(data_flat, storage_flat):
             store[cursor] = datum
 
+    @compile_disable()
     def _get_new_len(self, data, cursor):
         int_cursor = _is_int(cursor)
         ndim = self.ndim - int_cursor