adapted the tests to the changes in the code

tests.py

@@ -1,5 +1,4 @@
 import os
-import shutil
 
 import pytest as pytest
 from grouped_sampling import GroupedSamplingPipeLine
@@ -29,9 +28,6 @@ def test_get_supported_model_names():
 
 
 def test_on_server_start():
-    if os.path.exists(HUGGING_FACE_CACHE_DIR):
-        shutil.rmtree(HUGGING_FACE_CACHE_DIR)
-    assert not os.path.exists(HUGGING_FACE_CACHE_DIR)
     download_useful_models()
     assert os.path.exists(HUGGING_FACE_CACHE_DIR)
     assert len(os.listdir(HUGGING_FACE_CACHE_DIR)) > 0
@@ -64,10 +60,7 @@ def test_create_pipeline(model_name: str):
     assert pipeline is not None
     assert pipeline.model_name == model_name
     assert pipeline.wrapped_model.group_size == 5
-    assert pipeline.wrapped_model.end_of_sentence_stop is True
-    model_folder_name = "models--" + model_name.replace("/", "--")
-    model_path = os.path.join(HUGGING_FACE_CACHE_DIR, model_folder_name)
-    shutil.rmtree(model_path)
+    assert pipeline.wrapped_model.end_of_sentence_stop is False
 
 
 if __name__ == "__main__":