fix warnings

rwth-i6 · Feb 3, 2023 · fc861be · fc861be
1 parent eb46884
commit fc861be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/returnn/tf/layers/rec.py b/returnn/tf/layers/rec.py
@@ -972,8 +972,8 @@ def _get_output_cell(self, cell):
                 )
             self._last_hidden_state = final_state
         elif rnn_contrib and isinstance(
-            cell, (rnn_contrib.FusedRNNCell, rnn_contrib.LSTMBlockWrapper)
-        ):  # noqa # e.g. LSTMBlockFusedCell
+            cell, (rnn_contrib.FusedRNNCell, rnn_contrib.LSTMBlockWrapper)  # noqa # e.g. LSTMBlockFusedCell
+        ):
             # Will get (time,batch,ydim).
             assert self._max_seq_len is None
             y, final_state = cell(