From 51eab01fac3ffccea0ba3629ae868ad5ab3a2313 Mon Sep 17 00:00:00 2001
From: Chaoming Wang <adaduo@outlook.com>
Date: Thu, 22 Feb 2024 13:08:21 +0800
Subject: [PATCH] add `brainpy.math.VariableStack`

---
 brainpy/math/oo_transform.py            | 4 ++++
 docs/apis/brainpy.math.oo_transform.rst | 1 +
 2 files changed, 5 insertions(+)

diff --git a/brainpy/math/oo_transform.py b/brainpy/math/oo_transform.py
index 548a987d0..7654731d8 100644
--- a/brainpy/math/oo_transform.py
+++ b/brainpy/math/oo_transform.py
@@ -59,3 +59,7 @@
   eval_shape as eval_shape,
 )
 
+from brainpy._src.math.object_transform.variables import (
+  VariableStack as VariableStack,
+)
+
diff --git a/docs/apis/brainpy.math.oo_transform.rst b/docs/apis/brainpy.math.oo_transform.rst
index 754e0d81d..9ed9cf46a 100644
--- a/docs/apis/brainpy.math.oo_transform.rst
+++ b/docs/apis/brainpy.math.oo_transform.rst
@@ -77,4 +77,5 @@ Helpers for Object-oriented Transformations
    :template: classtemplate.rst
 
    eval_shape
+   VariableStack