amend

pytorch · Sep 14, 2022 · 8a45602 · 8a45602
1 parent ec41df6
commit 8a45602
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/envs/vec_env.py b/torchrl/envs/vec_env.py
@@ -20,7 +20,7 @@
 from torchrl import _check_for_faulty_process
 from torchrl.data import TensorDict, TensorSpec, CompositeSpec
 from torchrl.data.tensordict.tensordict import TensorDictBase, LazyStackedTensorDict
-from torchrl.data.utils import CloudpickleWrapper, DEVICE_TYPING
+from torchrl.data.utils import DEVICE_TYPING
 from torchrl.envs.common import EnvBase
 from torchrl.envs.env_creator import get_env_metadata, EnvCreator