bump + ruff

Author: ClashLuke

heavyball/utils.py

@@ -2921,8 +2921,6 @@ def _mv(x):
     return cond(x_norm > 0, _inner, lambda: x_norm.squeeze().clone()).squeeze()
 
 
-
-
 @decorator_knowngood
 def clamped_max_singular_value(
     A: Tensor, min: float, max_svd: int = 0, use_cholesky: bool = False, power_iter: int = 16
@@ -3136,7 +3134,7 @@ def psgd_pro_update_precond(
     lower_bount_beta: float,
     power_iter: int,
     dampening: float,
-max_step_size: float = 1/8
+    max_step_size: float = 1 / 8,
 ) -> None:
     """Update Kronecker product preconditioner Q with Q0.5EQ1.5 (PRO) method."""
     psgd_balance_Q(Q)

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "heavyball"
 description = "Compile-first PyTorch optimizer library - AdamW, Muon, SOAP/Shampoo, PSGD, Schedule-Free, and 30+ more with torch.compile fusion and composable features"
-version = "3.0.0"
+version = "3.0.1"
 authors = [{ name = "HeavyBall Authors", email = "github.heavyball@nestler.sh" }]
 classifiers = ["Intended Audience :: Developers",
     "Intended Audience :: Science/Research",