From c1d7cfee8f643b0138e962c3c3ad8966cef57d96 Mon Sep 17 00:00:00 2001 From: Jensun Ravichandran Date: Tue, 6 Jul 2021 17:07:26 +0200 Subject: [PATCH] fix(test): fix broken `CSVDataset` test --- tests/test_datasets.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index 66fd11e..5766c31 100644 --- 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):