From 724855d2511c1a6576cccada0aa0cb3b49207603 Mon Sep 17 00:00:00 2001
From: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com>
Date: Tue, 17 Dec 2024 00:14:32 +0000
Subject: [PATCH] Update NNX Variable class docs in variablelib.py

---
 flax/nnx/variablelib.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/flax/nnx/variablelib.py b/flax/nnx/variablelib.py
index 4752a9b7bd..11e3f33d01 100644
--- a/flax/nnx/variablelib.py
+++ b/flax/nnx/variablelib.py
@@ -54,10 +54,10 @@ class VariableMetadata(tp.Generic[A]):
 
 
 class Variable(tp.Generic[A], reprlib.Representable):
-  """The base class for all ``Variable`` types. Create custom ``Variable``
-  types by subclassing this class. Numerous NNX graph functions can filter
-  for specific ``Variable`` types, for example, :func:`split`, :func:`state`,
-  :func:`pop`, and :func:`State.filter`.
+  """The base class for all :class:`flax.nnx.Variable` types. Creates custom
+  ``nnx.Variable`` types by subclassing this class. Numerous NNX graph functions
+  can filter for specific ``nnx.Variable`` types, for example, :func:`flax.nnx.split`,
+  :func:`flax.nnx.state`, :func:`flax.nnx.pop`, and :func:`flax.nnx.State.filter`.
 
   Example usage::