amend

pytorch · Apr 24, 2024 · 1568b28 · 1568b28
1 parent 49b015d
commit 1568b28
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
@@ -29,7 +29,7 @@
 from torch import multiprocessing as mp
 from torch.utils._pytree import LeafSpec, tree_flatten, tree_map, tree_unflatten
 
-from torchrl._utils import _CKPT_BACKEND, implement_for, logger as torchrl_logger
+from torchrl._utils import implement_for, logger as torchrl_logger
 from torchrl.data.replay_buffers.utils import _is_int, INT_CLASSES
 
 SINGLE_TENSOR_BUFFER_NAME = os.environ.get(