fixed error

Author: Alexander

tests/test_loss_scaling.py

@@ -45,7 +45,7 @@ def test_scale_unscale(self):
     def test_adjust_finite_grads(self):
         """Test loss scaling adjustment with finite gradients"""
         # Test multiple periods to ensure scaling increases
-        scaling = self.loss_scalin
+        scaling = self.loss_scaling
         for _ in range(2000):  # One full period
             scaling = scaling.adjust(jnp.array(True))