init

pytorch · Jan 17, 2024 · 4502785 · 4502785
1 parent e5b07b8
commit 4502785
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/test_tensordict.py b/test/test_tensordict.py
@@ -2620,6 +2620,11 @@ def test_lock_write(self, td_name, device):
                 del td[key]
             td.lock_()
         else:
+            if td_name in (
+                "sub_td", "sub_td2", "permute_td", "squeezed_td",
+                "unsqueezed_td"):
+                # we can't call select inplace on these guys so we exit here
+                return
             with td.unlock_() if td.is_locked else contextlib.nullcontext():
                 td = td.select(inplace=True)
         for key, item in td_clone.items(True):