diff --git a/hyperlora/training_utils.py b/hyperlora/training_utils.py
index 6a6f4b3..d54f8cd 100644
--- a/hyperlora/training_utils.py
+++ b/hyperlora/training_utils.py
@@ -1,6 +1,6 @@
-from collections import defaultdict
 import json
 import logging
+from collections import defaultdict
 from dataclasses import fields
 from enum import Enum