julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(test): fix broken CSVDataset test

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-07-06 17:07:26 +02:00
parent 99be965581
commit c1d7cfee8f
No known key found for this signature in database
GPG Key ID: 3331B0F18B6D4D93
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_datasets.py
View File

@ -54,6 +54,8 @@ class TestCSVDataset(unittest.TestCase):
data = np.random.rand(100, 4) data = np.random.rand(100, 4)
targets = np.random.randint(2, size=(100, 1)) targets = np.random.randint(2, size=(100, 1))
arr = np.hstack([data, targets]) arr = np.hstack([data, targets])
if not os.path.exists("./artifacts"):
os.mkdir("./artifacts")
np.savetxt("./artifacts/test.csv", arr, delimiter=",") np.savetxt("./artifacts/test.csv", arr, delimiter=",")
def test_len(self): def test_len(self):