From 9228f8a194f7786c006b3d5272a567e98c08d675 Mon Sep 17 00:00:00 2001 From: jheek Date: Wed, 27 Oct 2021 10:27:50 +0200 Subject: [PATCH] Update flax/struct.py Co-authored-by: Marc van Zee --- flax/struct.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flax/struct.py b/flax/struct.py index c95d03fb9b..b4da6847a2 100644 --- a/flax/struct.py +++ b/flax/struct.py @@ -83,7 +83,7 @@ def __apply__(self, *args): valid containers that work with JAX transformations and more generally the `jax.tree_util` library. - Somtimes a "smart constructor" is desired, for example because + Sometimes a "smart constructor" is desired, for example because some of the attributes can be (optionally) derived from others. The way to do this with Flax dataclasses is to make a static or class method that provides the smart constructor.