From fb420e274e1c7861133b5822296949ba231b5a5c Mon Sep 17 00:00:00 2001
From: Cristian Garcia <cgarcia.e88@gmail.com>
Date: Thu, 16 Jun 2022 10:01:38 -0500
Subject: [PATCH] update version + changelog

---
 CHANGELOG.md    | 4 ++++
 README.md       | 2 +-
 flax/version.py | 2 +-
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 14be06c5d..0cf1fa166 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -26,6 +26,10 @@ vNext
 -
 -
 
+0.5.2
+-----
+- Fixes missing PyYAML dependecy.
+
 0.5.1
 -----
 New features:
diff --git a/README.md b/README.md
index b48bcc6bf..4521d52eb 100644
--- a/README.md
+++ b/README.md
@@ -190,7 +190,7 @@ To cite this repository:
   author = {Jonathan Heek and Anselm Levskaya and Avital Oliver and Marvin Ritter and Bertrand Rondepierre and Andreas Steiner and Marc van {Z}ee},
   title = {{F}lax: A neural network library and ecosystem for {JAX}},
   url = {http://github.com/google/flax},
-  version = {0.5.1},
+  version = {0.5.2},
   year = {2020},
 }
 ```
diff --git a/flax/version.py b/flax/version.py
index 108bc6ea0..3987ba3b7 100644
--- a/flax/version.py
+++ b/flax/version.py
@@ -13,5 +13,5 @@
 # limitations under the License.
 
 """Current Flax version at head on Github."""
-__version__ = "0.5.1"
+__version__ = "0.5.2"