Update

[ghstack-poisoned]
pytorch · Sep 25, 2024 · 94d5626 · 94d5626
1 parent a00ea10
commit 94d5626
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
                 ):