diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 61b3fbba..714d73a4 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -190,6 +190,7 @@ RUN pip install pysal \ seaborn python-dateutil dask python-igraph \ pyyaml joblib husl geopy mne pyshp \ pandas \ + polars \ flax && \ # Install h2o from source. # Use `conda install -c h2oai h2o` once Python 3.7 version is released to conda. diff --git a/tests/test_polars.py b/tests/test_polars.py new file mode 100644 index 00000000..c81a0b80 --- /dev/null +++ b/tests/test_polars.py @@ -0,0 +1,10 @@ +import unittest + +import polars as pl + +class TestPolars(unittest.TestCase): + def test_read_csv(self): + data = pl.read_csv("/input/tests/data/train.csv") + + self.assertEqual(100, len(data)) +