[5541172] update based on review comments

Signed-off-by: unknown <[email protected]>

Author: ynankani

examples/windows/accuracy_benchmark/kl_divergence_metrics/extract_logits_hf.py

@@ -5,7 +5,6 @@
 """
 
 import argparse
-import os
 import pickle
 import time
 
@@ -182,8 +181,6 @@ def main():
     global DEBUG
     DEBUG = args.debug
 
-
-
     try:
         # Extract logits
         start_time = time.time()