Merge branch 'main' into add-recipe-check-vllm-e2e

vllm-project · Nov 22, 2024 · e41d025 · e41d025
2 parents cfa6c76 + 19027b2
commit e41d025
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/e2e/vLLM/test_vllm.py b/tests/e2e/vLLM/test_vllm.py
@@ -102,6 +102,7 @@ def setUp(self):
             "The president of the US is",
             "My name is",
         ]
+        self.session = active_session()
 
     def test_vllm(self):
         import torch
@@ -153,6 +154,9 @@ def test_vllm(self):
         # check that expected files exist
         self._check_save_dir_has_expected_files()
 
+        # Reset after session info is extracted on save -- recipe
+        self.session.reset()
+
         # Run vLLM with saved model
         logger.info("================= RUNNING vLLM =========================")
         sampling_params = SamplingParams(temperature=0.80, top_p=0.95)