fix(test): fix broken CSVDataset test

2021-07-06 17:07:26 +02:00 · 2021-07-06 17:07:26 +02:00 · c1d7cfee8f
commit c1d7cfee8f
parent 99be965581
1 changed files with 2 additions and 0 deletions
--- a/tests/test_datasets.py
+++ b/tests/test_datasets.py
@ -54,6 +54,8 @@ class TestCSVDataset(unittest.TestCase):
        data = np.random.rand(100, 4)
        targets = np.random.randint(2, size=(100, 1))
        arr = np.hstack([data, targets])
+        if not os.path.exists("./artifacts"):
+            os.mkdir("./artifacts")
        np.savetxt("./artifacts/test.csv", arr, delimiter=",")

    def test_len(self):