From 99ac52565438f72a576897a58fed467fd2244708 Mon Sep 17 00:00:00 2001
From: ASEM000 <mahmoudasem00@gmail.com>
Date: Tue, 25 Jul 2023 23:49:54 +0900
Subject: [PATCH] doc edit req bump

---
 docs/requirements.txt | 2 +-
 serket/nn/state.py    | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/docs/requirements.txt b/docs/requirements.txt
index afb955e..ecc5841 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -18,5 +18,5 @@ sphinxcontrib-katex>=0.9.4
 typing-extensions>=4.5.0
 sphinx-copybutton>=0.5.1
 graphviz>=0.20.1
-pytreeclass>=0.4.0
+pytreeclass>=0.5.0
 kernex>=0.2.0
\ No newline at end of file
diff --git a/serket/nn/state.py b/serket/nn/state.py
index a923d42..d1cd991 100644
--- a/serket/nn/state.py
+++ b/serket/nn/state.py
@@ -55,8 +55,8 @@ def tree_state(tree: T, array: jax.Array | None = None) -> T:
         >>> tree = [1, 2, sk.nn.BatchNorm(5)]
         >>> sk.tree_state(tree)
         [NoState(), NoState(), BatchNormState(
-          running_mean=f32[5](μ=0.00, σ=0.00, ∈(0.00,0.00)),
-          running_var=f32[5](μ=1.00, σ=0.00, ∈(1.00,1.00))
+          running_mean=f32[5](μ=0.00, σ=0.00, ∈[0.00,0.00]),
+          running_var=f32[5](μ=1.00, σ=0.00, ∈[1.00,1.00])
         )]
 
     Example: