fix: remove .mindnlp in teardown (mindspore-lab#549)

qurikuduo · Jun 1, 2023 · 96cc1c9 · 96cc1c9
1 parent 6c93d38
commit 96cc1c9
Show file tree

Hide file tree

Showing 22 changed files with 140 additions and 0 deletions.
diff --git a/tests/ut/models/bert/test_modeling_bert.py b/tests/ut/models/bert/test_modeling_bert.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test Bert"""
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -69,3 +70,8 @@ def test_from_pretrained_from_pt(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/bloom/test_bloom_config.py b/tests/ut/models/bloom/test_bloom_config.py
@@ -15,6 +15,7 @@
 """Test Bloom Config"""
 
 import gc
+import os
 import unittest
 import pytest
 
@@ -36,3 +37,8 @@ def test_bloom_config(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/bloom/test_modeling_bloom.py b/tests/ut/models/bloom/test_modeling_bloom.py
@@ -15,6 +15,7 @@
 """Test Bloom"""
 
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -57,3 +58,8 @@ def test_from_pretrained_from_pt(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/clip/test_modeling_clip.py b/tests/ut/models/clip/test_modeling_clip.py
@@ -13,6 +13,8 @@
 # limitations under the License.
 # ============================================================================
 """Test CLIP"""
+import gc
+import os
 import unittest
 import numpy as np
 import mindspore
@@ -234,3 +236,11 @@ def test_clip_vision_model_with_projection(self):
 
         assert outputs[0].shape == (1, self.text_config.hidden_size)
         assert outputs[1].shape == (1, 50, self.vision_config.hidden_size)
+
+    def tearDown(self) -> None:
+        gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/codegen/test_modeling_codegen.py b/tests/ut/models/codegen/test_modeling_codegen.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test CodeGen"""
 import gc
+import os
 import unittest
 
 import mindspore
@@ -97,3 +98,8 @@ def test_codegen_forcausallm(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/cpm/test_modeling_cpm_ant.py b/tests/ut/models/cpm/test_modeling_cpm_ant.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test CPM Ant"""
 import gc
+import os
 import unittest
 import numpy as np
 
@@ -60,3 +61,8 @@ def test_cpm_ant_for_causal_lm(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/ernie/test_ernie.py b/tests/ut/models/ernie/test_ernie.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test Ernie"""
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -170,3 +171,8 @@ def test_from_pretrained(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/glm/test_modeling_glm.py b/tests/ut/models/glm/test_modeling_glm.py
@@ -13,6 +13,8 @@
 # limitations under the License.
 # ============================================================================
 """Test GLM"""
+import gc
+import os
 import unittest
 import numpy as np
 
@@ -137,3 +139,11 @@ def test_glm_model(self):
         output_tuple = model(input_ids_input, position_ids_input, attention_mask_input, mems)
         output = output_tuple[0]
         assert output.shape == (batch_size, max_sequence_length, hidden_size)
+
+    def tearDown(self) -> None:
+        gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/gpt/test_modeling_gpt.py b/tests/ut/models/gpt/test_modeling_gpt.py
@@ -17,6 +17,7 @@
 # ============================================================================
 """Test GPT"""
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -159,3 +160,8 @@ def test_from_pretrained_from_pt(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/gpt2/test_modeling_gpt2.py b/tests/ut/models/gpt2/test_modeling_gpt2.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test GPT2"""
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -152,3 +153,8 @@ def test_from_pretrained_from_pt(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/gpt_neo/test_gpt_neo.py b/tests/ut/models/gpt_neo/test_gpt_neo.py
@@ -15,6 +15,7 @@
 # limitations under the License.
 """Test GPTNeo"""
 import gc
+import os
 import unittest
 import numpy as np
 import mindspore
@@ -146,3 +147,8 @@ def test_gptneo_for_sequence_classification(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/llama/test_modeling_llama.py b/tests/ut/models/llama/test_modeling_llama.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test Llama"""
 import gc
+import os
 import unittest
 import numpy as np
 import mindspore
@@ -109,3 +110,8 @@ def test_llama_transformer(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/llama/test_modeling_llama_hf.py b/tests/ut/models/llama/test_modeling_llama_hf.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test Llama"""
 import gc
+import os
 import unittest
 import numpy as np
 import mindspore
@@ -142,3 +143,8 @@ def test_llama_for_sequence_classification(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/longformer/test_embeddings.py b/tests/ut/models/longformer/test_embeddings.py
@@ -15,6 +15,7 @@
 # ============================================================================
 """Test Longformer"""
 import gc
+import os
 import unittest
 import numpy as np
 import mindspore
@@ -360,3 +361,8 @@ def test_modeling_longformer_multiple_choice(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/luke/test_modeling_luke.py b/tests/ut/models/luke/test_modeling_luke.py
@@ -17,6 +17,7 @@
 # pylint:disable=R0904
 """Test LUKE"""
 import gc
+import os
 import unittest
 
 import mindspore
@@ -265,3 +266,8 @@ def test_luke_for_multiple_choice(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/mobilebert/test_mobilebert.py b/tests/ut/models/mobilebert/test_mobilebert.py
@@ -19,6 +19,7 @@
 Test MobileBert
 """
 import gc
+import os
 import unittest
 
 import mindspore
@@ -340,3 +341,8 @@ def test_mobilebert_fortokenclassification(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/nezha/test_modeling_nezha.py b/tests/ut/models/nezha/test_modeling_nezha.py
@@ -16,6 +16,7 @@
 Test Nezha
 """
 import gc
+import os
 import unittest
 import random
 import numpy as np
@@ -278,3 +279,8 @@ def test_for_question_answering(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/roberta/test_modeling_roberta.py b/tests/ut/models/roberta/test_modeling_roberta.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test Roberta"""
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -72,3 +73,8 @@ def test_from_pretrained(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/rwkv/test_modeling_rwkv.py b/tests/ut/models/rwkv/test_modeling_rwkv.py
@@ -14,6 +14,7 @@
 # ============================================================================
 """Test RWKV"""
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -53,3 +54,8 @@ def forward(input_ids):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/t5/test_modeling_t5.py b/tests/ut/models/t5/test_modeling_t5.py
@@ -15,6 +15,7 @@
 # pylint:disable=W0212
 """Test T5"""
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -224,3 +225,8 @@ def test_from_pretrained_from_pt(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/tinybert/test_tinybert.py b/tests/ut/models/tinybert/test_tinybert.py
@@ -17,6 +17,7 @@
 Test TinyBert
 """
 import gc
+import os
 import unittest
 
 import numpy as np
@@ -365,3 +366,8 @@ def test_tiny_bert_for_sequence_classification(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")
diff --git a/tests/ut/models/xlm/test_xlm.py b/tests/ut/models/xlm/test_xlm.py
@@ -16,6 +16,7 @@
 Test XLM
 """
 import gc
+import os
 import unittest
 import pytest
 import numpy as np
@@ -181,3 +182,8 @@ def test_from_pretrained_from_pt(self):
 
     def tearDown(self) -> None:
         gc.collect()
+
+    @classmethod
+    def tearDownClass(cls):
+        if os.path.exists("~/.mindnlp"):
+            os.removedirs("~/.mindnlp")