From 8aa8c0ddc62922e017a2f136f74fd08571ecf3ea Mon Sep 17 00:00:00 2001 From: Marco Edward Gorelli <33491632+MarcoGorelli@users.noreply.github.com> Date: Thu, 9 Feb 2023 17:39:27 +0000 Subject: [PATCH] Add polars https://www.pola.rs/ This one's well good, and is growing in popularity --- Dockerfile.tmpl | 1 + tests/test_polars.py | 10 ++++++++++ 2 files changed, 11 insertions(+) create mode 100644 tests/test_polars.py 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)) +