lint

Signed-off-by: Jennifer Chen <[email protected]>

Author: jenchen13

examples/nemo_run/qat/nemo_qat_flow.py

@@ -138,7 +138,7 @@ def get_args():
         "--enable_kv_cache",
         help="Enables KV-cache quantization",
         action="store_true",
-        default=False
+        default=False,
     )
     return parser.parse_args()
 

modelopt/torch/export/plugins/nemo_run.py

@@ -57,7 +57,7 @@ def _get_most_recent_ckpt(directory: str):
         str: Path to the most recent subdirectory.
     """
     exp_dir = Path(directory) / "default"
-    if not exp_dir.exists(): 
+    if not exp_dir.exists():
         raise FileNotFoundError(f"Experiment directory {exp_dir} does not exist")
 
     checkpoint_dir = exp_dir / "checkpoints"