[BugFix] Temporarily set lazy legacy to True (pytorch#1840)

ai4co · Jan 26, 2024 · 6a42116 · 6a42116
1 parent 6f90397
commit 6a42116
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/torchrl/__init__.py b/torchrl/__init__.py
@@ -7,8 +7,12 @@
 
 import torch
 
+from tensordict import set_lazy_legacy
+
 from torch import multiprocessing as mp
 
+set_lazy_legacy(True).set()
+
 if torch.cuda.device_count() > 1:
     n = torch.cuda.device_count() - 1
     os.environ["MUJOCO_EGL_DEVICE_ID"] = str(1 + (os.getpid() % n))