From 1568b2897e5d2436083657c239fe4b394a9b738e Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Wed, 24 Apr 2024 21:30:54 +0100 Subject: [PATCH] amend --- torchrl/data/replay_buffers/storages.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchrl/data/replay_buffers/storages.py b/torchrl/data/replay_buffers/storages.py index a036b39103e..a57c0fc94f2 100644 --- 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(