Added stop torchserve to unit tests

pytorch · Jun 3, 2024 · 1ce8bf0 · 1ce8bf0
1 parent 5f87195
commit 1ce8bf0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/test/pytest/test_example_stateful_sequence_batching_http.py b/test/pytest/test_example_stateful_sequence_batching_http.py
@@ -131,6 +131,7 @@ def test_stateful_mar(mar_file_path, model_store):
 
         # Clean up files
         shutil.rmtree(Path(model_store) / model_name)
+        test_utils.stop_torchserve()
 
 
 def __infer_stateful(model_name, sequence_id, expected):

diff --git a/test/pytest/test_example_stateful_sequence_continuous_batching_http.py b/test/pytest/test_example_stateful_sequence_continuous_batching_http.py
@@ -137,6 +137,7 @@ def test_infer_stateful(mar_file_path, model_store):
 
         # Clean up files
         shutil.rmtree(Path(model_store) / model_name)
+        test_utils.stop_torchserve()
 
 
 def test_infer_stateful_end(mar_file_path, model_store):
@@ -191,6 +192,7 @@ def test_infer_stateful_end(mar_file_path, model_store):
 
         # Clean up files
         shutil.rmtree(Path(model_store) / model_name)
+        test_utils.stop_torchserve()
 
 
 def test_infer_stateful_cancel(mar_file_path, model_store):
@@ -255,6 +257,7 @@ def test_infer_stateful_cancel(mar_file_path, model_store):
 
         # Clean up files
         shutil.rmtree(Path(model_store) / model_name)
+        test_utils.stop_torchserve()
 
 
 def __infer_stateful(model_name, sequence_id, expected):