From eb41b61aa0d5cf144a65e487314c914f0498389e Mon Sep 17 00:00:00 2001 From: vmoens Date: Thu, 8 Jan 2026 10:01:04 +0000 Subject: [PATCH] Update [ghstack-poisoned] --- torchrl/objectives/crossq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchrl/objectives/crossq.py b/torchrl/objectives/crossq.py index 68c7881c6dc..0c2f55258b8 100644 --- a/torchrl/objectives/crossq.py +++ b/torchrl/objectives/crossq.py @@ -374,8 +374,8 @@ def maybe_init_target_entropy(self, fault_tolerant=True): if "_target_entropy" in self._buffers: return target_entropy = self._target_entropy + device = next(self.parameters()).device if target_entropy == "auto": - device = next(self.parameters()).device action_spec = self.get_action_spec() if action_spec is None: if fault_tolerant: