Update

[ghstack-poisoned]

torchrl/collectors/collectors.py

@@ -189,7 +189,7 @@ def _get_policy_and_device(
                     "The collector will trust that the devices match. To suppress this "
                     "warning, set `trust_policy=True` when building the collector."
                 )
-                return policy, None
+            return policy, None
 
         def map_weight(
             weight,