Skip to content

Commit 8682b1b

Browse files
authored
Add vaex package. (#1079)
Fixes #928. Original PR is #1031 (thanks @JovanVeljanoski) - Includes a test. http://b/191699352
1 parent a6695cb commit 8682b1b

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

Dockerfile.tmpl

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

408408
RUN pip install flashtext && \
409409
pip install wandb && \
410+
pip install vaex && \
410411
pip install marisa-trie && \
411412
pip install pyemd && \
412413
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)