pre-commit lint fixes

Author: ryan-williams

src/haliax/nn/linear.py

@@ -159,12 +159,14 @@ def _out_first(self):
 
     def to_state_dict(self, prefix: Optional[str] = None) -> StateDict:
         # weight can be None for certain filtering things like LoRA
-        scaled = dataclasses.replace(self, weight=self.weight * self.reparam.active_scale if self.weight is not None else None)
+        scaled = dataclasses.replace(
+            self, weight=self.weight * self.reparam.active_scale if self.weight is not None else None
+        )
         return default_eqx_module_to_state_dict(scaled, prefix)
 
     def from_state_dict(self: Mod, state_dict: StateDict, prefix: Optional[str] = None) -> Mod:
         unscaled = default_eqx_module_from_state_dict(self, state_dict, prefix)
-        if unscaled.weight is not None: 
+        if unscaled.weight is not None:
             unscaled = dataclasses.replace(unscaled, weight=unscaled.weight / self.reparam.active_scale)
         return unscaled