fix precommit

src/haliax/__init__.py

@@ -56,8 +56,8 @@
     unflatten_axis,
     updated_slice,
 )
-from .jax_utils import filter_checkpoint
 from .hof import fold, map, scan, vmap
+from .jax_utils import filter_checkpoint
 from .ops import clip, isclose, pad_left, trace, tril, triu, where
 from .partitioning import auto_sharded, axis_mapping, fsdp, named_jit, shard, shard_with_axis_mapping
 from .specialized_fns import top_k