diff --git a/lora/lora.py b/lora/lora.py
index 23a9afdc..6f91ccca 100644
--- a/lora/lora.py
+++ b/lora/lora.py
@@ -1,9 +1,9 @@
 # Copyright © 2023-2024 Apple Inc.
 
-import sys
 import argparse
 import json
 import math
+import sys
 import time
 from pathlib import Path
 
@@ -15,7 +15,6 @@ import utils as lora_utils
 from mlx.utils import tree_flatten
 from models import LoRALinear
 
-
 # Disable output buffering to see print statements in real-time
 sys.stdout.reconfigure(line_buffering=True)