amend

pytorch · Mar 23, 2024 · 69ac3d2 · 69ac3d2
1 parent 24d2a25
commit 69ac3d2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchrl/collectors/collectors.py b/torchrl/collectors/collectors.py
@@ -494,7 +494,7 @@ def __init__(
             elif torch.backends.mps.is_available():
                 self._sync_env = torch.mps.synchronize
             elif self.env_device.type == "cpu":
-                self._sync_storage = _do_nothing
+                self._sync_env = _do_nothing
             else:
                 raise RuntimeError("Non supported device")
         else:
@@ -507,7 +507,7 @@ def __init__(
             elif torch.backends.mps.is_available():
                 self._sync_policy = torch.mps.synchronize
             elif self.policy_device.type == "cpu":
-                self._sync_storage = _do_nothing
+                self._sync_policy = _do_nothing
             else:
                 raise RuntimeError("Non supported device")
         else: