From 372abd43be28b33b6555367ddf1159d5dce653d1 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Mon, 30 Sep 2024 11:48:57 +0100
Subject: [PATCH] Update

[ghstack-poisoned]
---
 torchrl/collectors/collectors.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchrl/collectors/collectors.py b/torchrl/collectors/collectors.py
index 1a9e445316d..4a83258278d 100644
--- a/torchrl/collectors/collectors.py
+++ b/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,