Merge pull request #3186 from google:fix-rnn-docs

PiperOrigin-RevId: 546391233

flax/linen/recurrent.py

@@ -50,7 +50,11 @@
 Carry = Any
 CarryHistory = Any
 Output = Any
-NEVER = object()
+
+class _Never:
+  pass
+
+NEVER = _Never()
 
 LEGACY_UPDATE_MESSAGE = (
   "The RNNCellBase API has changed, "
@@ -649,8 +653,6 @@ class RNN(Module):
 
   Attributes:
     cell: an instance of :class:`RNNCellBase`.
-    cell_size: the size of the cell as requested by :meth:`RNNCellBase.initialize_carry`,
-      it can be an integer or a tuple of integers.
     time_major: if ``time_major=False`` (default) it will expect inputs with shape
       ``(*batch, time, *features)``, else it will expect inputs with shape ``(time, *batch, *features)``.
     return_carry: if ``return_carry=False`` (default) only the output sequence is returned,