From 7c597d828a62d47e5b9e7058e458615119f37f1a Mon Sep 17 00:00:00 2001
From: Daniel Johnson <ddjohnson@google.com>
Date: Wed, 11 Sep 2024 10:18:56 -0700
Subject: [PATCH] Release v0.2.2

PiperOrigin-RevId: 673441884
---
 penzai/__init__.py | 2 +-
 pyproject.toml     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/penzai/__init__.py b/penzai/__init__.py
index 03bc5c8..d56dfd7 100644
--- a/penzai/__init__.py
+++ b/penzai/__init__.py
@@ -14,4 +14,4 @@
 
 """A JAX research toolkit for building, editing, and visualizing neural networks."""
 
-__version__ = '0.2.1'
+__version__ = '0.2.2'
diff --git a/pyproject.toml b/pyproject.toml
index 2fac10e..e41e1a3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,7 +23,7 @@ dependencies = [
     "jax>=0.4.23",
     "numpy>=1.25.2",
     "ordered_set>=4.1.0",
-    "treescope>=0.1.0",
+    "treescope>=0.1.3",
     "typing_extensions>=4.2",
 ]