[Testing] Reduce error verbosity of cleanup (#1365)

## Purpose ##
* When tests fail, they often show the error that occurs during cleanup,
rather than the error that caused the tests to fail
```
ERROR tests/llmcompressor/transformers/obcq/test_consecutive_runs.py::TestConsecutiveRunsSmall_0_commit::test_consecutive_runs_small - FileNotFoundError: [Errno 2] No such file or directory: './oneshot_output'
```

## Changes ## 
* Check if a directory exists before attempting to delete it during
cleanup

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

tests/e2e/vLLM/test_vllm.py

@@ -170,7 +170,7 @@ def test_vllm(self, test_data_file: str):
         self.tear_down()
 
     def tear_down(self):
-        if self.save_dir is not None:
+        if self.save_dir is not None and os.path.isdir(self.save_dir):
             shutil.rmtree(self.save_dir)
 
         timer = get_singleton_manager()

tests/llmcompressor/transformers/compression/test_decompress.py

@@ -1,4 +1,5 @@
 import copy
+import os
 import shutil
 import tempfile
 import unittest
@@ -128,7 +129,8 @@ def test_hf_quantizer_decompress_match_manual_decompress(self):
 
     @classmethod
     def tearDownClass(self):
-        shutil.rmtree(self.test_dir)
+        if os.path.isdir(self.test_dir):
+            shutil.rmtree(self.test_dir)
         del self.dense_model
         del self.decompressed_model_hf_quantizer
         del self.decompressed_model_manual

tests/llmcompressor/transformers/compression/test_quantization.py

@@ -49,7 +49,8 @@ def setUpClass(cls):
 
     @classmethod
     def tearDownClass(cls):
-        shutil.rmtree(cls.test_dir)
+        if os.path.isdir(cls.test_dir):
+            shutil.rmtree(cls.test_dir)
         del cls.model
         torch.cuda.empty_cache()
 

tests/llmcompressor/transformers/compression/test_run_compressed.py

@@ -1,3 +1,4 @@
+import os
 import shutil
 import tempfile
 import unittest
@@ -82,7 +83,8 @@ def test_compressed_matches_decompressed(self):
 
     @classmethod
     def tearDownClass(cls):
-        shutil.rmtree(cls.test_dir)
+        if os.path.isdir(cls.test_dir):
+            shutil.rmtree(cls.test_dir)
         del cls.decompressed_model
         del cls.uncompressed_model
         torch.cuda.empty_cache()
@@ -167,7 +169,8 @@ def test_compressed_matches_decompressed__hf_quantizer(self):
 
     @classmethod
     def tearDownClass(cls):
-        shutil.rmtree(cls.test_dir)
+        if os.path.isdir(cls.test_dir):
+            shutil.rmtree(cls.test_dir)
         del cls.decompressed_model
         del cls.compressed_model
         torch.cuda.empty_cache()

tests/llmcompressor/transformers/finetune/test_finetune_no_recipe_custom_dataset.py

@@ -107,7 +107,8 @@ def create_mock_file(self, extension, content, path, filename):
         return mock_filepath  # Return the file path
 
     def tearDown(self):
-        shutil.rmtree(self.output)
+        if os.path.isdir(self.output):
+            shutil.rmtree(self.output)
 
 
 @pytest.mark.integration

tests/llmcompressor/transformers/finetune/test_finetune_oneshot_with_modifier.py

@@ -1,3 +1,4 @@
+import os
 import shutil
 import unittest
 from pathlib import Path
@@ -46,4 +47,5 @@ def test_oneshot_with_modifier_object(self):
         )
 
     def tearDown(self):
-        shutil.rmtree(self.output)
+        if os.path.isdir(self.output):
+            shutil.rmtree(self.output)

tests/llmcompressor/transformers/finetune/test_finetune_without_recipe.py

@@ -1,3 +1,4 @@
+import os
 import shutil
 import unittest
 
@@ -41,4 +42,5 @@ def test_finetune_without_recipe(self):
         )
 
     def tearDown(self):
-        shutil.rmtree(self.output)
+        if os.path.isdir(self.output):
+            shutil.rmtree(self.output)

tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune.py

@@ -75,7 +75,8 @@ def _test_oneshot_and_finetune(self):
     def tearDown(self):
         # TODO: we get really nice stats from finetune that we should log
         # stored in results.json
-        shutil.rmtree(self.output)
+        if os.path.isdir(self.output):
+            shutil.rmtree(self.output)
 
 
 @pytest.mark.integration

tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune_with_tokenizer.py

@@ -1,3 +1,4 @@
+import os
 import shutil
 import unittest
 
@@ -78,4 +79,5 @@ def test_oneshot_and_finetune_with_tokenizer(self):
         print(tokenizer.decode(output[0]))
 
     def tearDown(self):
-        shutil.rmtree(self.output)
+        if os.path.isdir(self.output):
+            shutil.rmtree(self.output)

tests/llmcompressor/transformers/finetune/test_oneshot_then_finetune.py

@@ -1,3 +1,4 @@
+import os
 import shutil
 import unittest
 from pathlib import Path
@@ -175,4 +176,5 @@ def test_oneshot_quantization_then_finetune(self):
             )
 
     def tearDown(self):
-        shutil.rmtree(self.output)
+        if os.path.isdir(self.output):
+            shutil.rmtree(self.output)