clean up test

Signed-off-by: Max Lepikhin <[email protected]>
opensearch-project · Dec 12, 2024 · 2e6a0a3 · 2e6a0a3
1 parent f6d611e
commit 2e6a0a3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/ml-algorithms/src/test/java/org/opensearch/ml/engine/utils/FileUtilsTest.java b/ml-algorithms/src/test/java/org/opensearch/ml/engine/utils/FileUtilsTest.java
@@ -22,16 +22,19 @@
 import org.junit.rules.TemporaryFolder;
 
 public class FileUtilsTest {
-    private final TemporaryFolder tempDir = new TemporaryFolder();
+    private TemporaryFolder tempDir;
 
     @Before
     public void setUp() throws Exception {
+        tempDir = new TemporaryFolder();
         tempDir.create();
     }
 
     @After
-    public void tearUp() {
-        tempDir.delete();
+    public void tearDown() {
+        if (tempDir != null) {
+            tempDir.delete();
+        }
     }
 
     @Test