From d8d1cac422c13e604c2c42867432152dd0c7db9e Mon Sep 17 00:00:00 2001
From: Vincent Moens <vincentmoens@gmail.com>
Date: Tue, 6 Feb 2024 09:15:41 +0000
Subject: [PATCH] [BugFix] state typo in RNG control module

---
 torchrl/_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchrl/_utils.py b/torchrl/_utils.py
index 6c52b1d66e7..ae01556f0e6 100644
--- 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: