Skip to content

Commit 3e6de39

Browse files
authored
[BugFix] Fix compat with pt 2.1.1 (#3157)
1 parent 0a17d6d commit 3e6de39

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchrl/data/replay_buffers/storages.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@
5050
try:
5151
from torch.compiler import is_compiling
5252
except ImportError:
53-
from torch._dynamo import is_dynamo_compiling as is_compiling
53+
from torch._dynamo import is_compiling
5454

5555

5656
class Storage:

0 commit comments

Comments
 (0)