doc edit req bump

ASEM000 · Jul 25, 2023 · 99ac525 · 99ac525
1 parent 37d950f
commit 99ac525
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
diff --git 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: