From e36fe1128cf8de1063f22cefd6e97279f20c4045 Mon Sep 17 00:00:00 2001
From: Tom Begley <tomcbegley@gmail.com>
Date: Fri, 25 Nov 2022 11:13:25 +0000
Subject: [PATCH] TensorDictSequential inheritance improvement

---
 tensordict/nn/sequence.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensordict/nn/sequence.py b/tensordict/nn/sequence.py
index 41a7d7cb1..15269a287 100644
--- a/tensordict/nn/sequence.py
+++ b/tensordict/nn/sequence.py
@@ -242,7 +242,7 @@ def select_subsequence(
                 "No modules left after selection. Make sure that in_keys and out_keys are coherent."
             )
 
-        return TensorDictSequential(*modules)
+        return self.__class__(*modules)
 
     def _run_module(
         self,