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: