Fix typo in struct.py documentation

google · Jul 18, 2023 · 0a4b0d6 · 0a4b0d6
1 parent 881c449
commit 0a4b0d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: