Update e2e test timings ouputs (#1179)

SUMMARY: - Update scheme for kv_cache test so that it saves unique files - Update pandas command to prevent it from adding an extra 0 column in the saved csv
vllm-project · Feb 21, 2025 · 1101723 · 1101723
1 parent 5a82dcc
commit 1101723
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/e2e/vLLM/configs/kv_cache_gptq_tinyllama.yaml b/tests/e2e/vLLM/configs/kv_cache_gptq_tinyllama.yaml
@@ -4,4 +4,4 @@ model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
 recipe: tests/e2e/vLLM/recipes/kv_cache/gptq.yaml
 dataset_id: HuggingFaceH4/ultrachat_200k
 dataset_split: train_sft
-scheme: kv_cache_default_tinyllama
+scheme: kv_cache_default_gptq_tinyllama
diff --git a/tests/e2e/vLLM/test_vllm.py b/tests/e2e/vLLM/test_vllm.py
@@ -182,7 +182,7 @@ def tear_down(self):
             p.mkdir(parents=True, exist_ok=True)
 
             df = pd.DataFrame(measurements)
-            df.to_csv(p / f"{self.save_dir}.csv")
+            df.to_csv(p / f"{self.save_dir}.csv", index=False)
 
     @log_time
     def _save_compressed_model(self, oneshot_model, tokenizer):