[BugFix] state typo in RNG control module

pytorch · Feb 6, 2024 · d8d1cac · d8d1cac
1 parent eec9f9e
commit d8d1cac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/_utils.py b/torchrl/_utils.py
@@ -729,7 +729,7 @@ def _get_state(self):
                 )
 
         else:
-            self.state = torch.random.get_rng_state()
+            self._state = torch.random.get_rng_state()
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         if self.has_cuda: