Skip to content

Commit 411e2f5

Browse files
Add the vaex library (includes a test) (#1031)
1 parent 6d08285 commit 411e2f5

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,7 @@ RUN pip install bleach && \
365365

366366
RUN pip install flashtext && \
367367
pip install wandb && \
368+
pip install vaex && \
368369
pip install marisa-trie && \
369370
pip install pyemd && \
370371
pip install pyupset && \

tests/test_vaex.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
import unittest
2+
3+
import vaex
4+
5+
class TestVaex(unittest.TestCase):
6+
def test_read_csv(self):
7+
df = vaex.read_csv("/input/tests/data/train.csv")
8+
9+
self.assertEqual((100, 785), df.shape)
10+
self.assertEqual(10, df['label'].nunique())

0 commit comments

Comments
 (0)