Merge pull request #3209 from marcoselvi:patch-1

PiperOrigin-RevId: 549447778
google · Jul 19, 2023 · ac6cde0 · ac6cde0
2 parents 54a0296 + 0a4b0d6
commit ac6cde0
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: