From 9e84a045ac2d6c2232b1b3ec51428d5f6934cba2 Mon Sep 17 00:00:00 2001 From: Albert Bou Date: Fri, 1 Dec 2023 11:24:28 +0100 Subject: [PATCH] Update torchrl/modules/tensordict_module/rnn.py Co-authored-by: Vincent Moens --- torchrl/modules/tensordict_module/rnn.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/torchrl/modules/tensordict_module/rnn.py b/torchrl/modules/tensordict_module/rnn.py index ef24b6c5a5c..1d356d46c1f 100644 --- a/torchrl/modules/tensordict_module/rnn.py +++ b/torchrl/modules/tensordict_module/rnn.py @@ -149,7 +149,10 @@ def lstm_cell(self, x, hx, cx): class LSTM(nn.LSTM): - """A module that runs multiple steps of a multi-layer LSTM and is only coded in Python.""" + """A module that runs multiple steps of a multi-layer LSTM and is only coded in Python. + + """ + __doc__ += nn.LSTM.__doc__ def __init__( self,