format adjustment

Author: xinSky00

examples/offline_inference_rerope.py

@@ -23,12 +23,11 @@
 def setup_environment_variables():
     os.environ["VLLM_USE_V1"] = "1"
     os.environ["PYTHONHASHSEED"] = "123456"
-    
+
     os.environ["VLLM_ATTENTION_BACKEND"] = "TRITON_ATTN_VLLM_V1"
     os.environ["REROPE_WINDOW"] = "32768"
     os.environ["TRAINING_LENGTH"] = "32768"
 
-
     global data_dir
     data_dir = os.getenv("DATA_DIR", "/home/data/kv_cache")
     data_dir = input(
@@ -124,9 +123,7 @@ def main():
             if not os.path.isfile(path_to_dataset):
                 print("Exiting. Incorrect dataset path")
                 sys.exit(1)
-        with open(
-            path_to_dataset, "r", encoding="utf-8"
-        ) as f:
+        with open(path_to_dataset, "r", encoding="utf-8") as f:
             for line in f:
                 data_all.append(json.loads(line))
 
@@ -194,4 +191,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-