Skip to content

Commit

Permalink
fix zip_strict import
Browse files Browse the repository at this point in the history
  • Loading branch information
vmoens committed Oct 9, 2024
1 parent 36936ed commit 684e15a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchrl/envs/batched_envs.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,13 +22,13 @@
import torch

from tensordict import (
_zip_strict,
is_tensor_collection,
LazyStackedTensorDict,
TensorDict,
TensorDictBase,
unravel_key,
)
from tensordict.utils import _zip_strict
from torch import multiprocessing as mp
from torchrl._utils import (
_check_for_faulty_process,
Expand Down

0 comments on commit 684e15a

Please sign in to comment.