[BugFix] Extend RB with lazy stack (revamp)

ghstack-source-id: df397d09166d8fb61eceacb5fe8659e0295ca414 Pull Request resolved: #2454
pytorch · Sep 25, 2024 · ca3a595 · ca3a595
1 parent 1aca00e
commit ca3a595
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/data/replay_buffers/storages.py b/torchrl/data/replay_buffers/storages.py
@@ -1118,7 +1118,7 @@ def max_size_along_dim0(data_shape):
             out = data.clone().to(self.device)
             out = out.expand(max_size_along_dim0(data.shape))
             out = out.memmap_like(prefix=self.scratch_dir, existsok=self.existsok)
-            if torchrl_logger.getEffectiveLevel() == logging.DEBUG:
+            if torchrl_logger.isEnabledFor(logging.DEBUG):
                 for key, tensor in sorted(
                     out.items(include_nested=True, leaves_only=True), key=str
                 ):