diff --git a/TrainingExtensions/torch/src/python/aimet_torch/_base/auto_quant.py b/TrainingExtensions/torch/src/python/aimet_torch/_base/auto_quant.py
index cfcda62cb54..1c22149e3d7 100644
--- a/TrainingExtensions/torch/src/python/aimet_torch/_base/auto_quant.py
+++ b/TrainingExtensions/torch/src/python/aimet_torch/_base/auto_quant.py
@@ -1395,5 +1395,3 @@ def _optimize_main_wrapper(fp32_model, target_acc):
         yield spy
     finally:
         setattr(auto_quant, "_optimize_main", _optimize_main)
-
-