amend

pytorch · Nov 21, 2023 · 7b26175 · 7b26175
1 parent ab88669
commit 7b26175
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensordict/_td.py b/tensordict/_td.py
@@ -1246,8 +1246,8 @@ def _set_str(
         if not validated:
             value = self._validate_value(value, check_shape=True)
         if not inplace:
-            # if self.is_locked:
-            #     raise RuntimeError(_LOCK_ERROR)
+            if self.is_locked:
+                raise RuntimeError(_LOCK_ERROR)
             self._tensordict[key] = value
         else:
             try: