From 0a4b0d68a358fb3b56d3ab91413b292ca70a65fc Mon Sep 17 00:00:00 2001 From: Marco Selvi Date: Tue, 18 Jul 2023 14:44:05 +0100 Subject: [PATCH] Fix typo in struct.py documentation --- flax/struct.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flax/struct.py b/flax/struct.py index f434593f7c..aeafdf4c15 100644 --- a/flax/struct.py +++ b/flax/struct.py @@ -86,7 +86,7 @@ class DirectionAndScaleKernel: @classmethod def create(cls, kernel): scale = jax.numpy.linalg.norm(kernel, axis=0, keepdims=True) - directin = direction / scale + direction = direction / scale return cls(direction, scale) Args: