From c5b058c51112566cb13681ecbd463c865533e55e Mon Sep 17 00:00:00 2001 From: DJ Sterling Date: Fri, 27 Sep 2024 11:29:19 -0600 Subject: [PATCH 1/2] Add hvplot --- Dockerfile.tmpl | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index c0037839..326343da 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -484,6 +484,7 @@ RUN rm -rf /opt/conda/lib/python3.10/site-packages/numpy* && \ s3fs \ gcsfs \ kaggle-environments \ + hvplot \ # geopandas > v0.14.4 breaks learn tools geopandas==v0.14.4 \ "shapely<2" \ From a762dbe4e2de06c3b3951016b90473dfc8f99201 Mon Sep 17 00:00:00 2001 From: DJ Sterling Date: Fri, 27 Sep 2024 12:05:50 -0600 Subject: [PATCH 2/2] add test to polars test --- tests/test_polars.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/test_polars.py b/tests/test_polars.py index c81a0b80..8526bd29 100644 --- a/tests/test_polars.py +++ b/tests/test_polars.py @@ -2,9 +2,14 @@ import polars as pl -class TestPolars(unittest.TestCase): +class TestPolars(unittest.TestCase): def test_read_csv(self): - data = pl.read_csv("/input/tests/data/train.csv") + data = pl.read_csv('/input/tests/data/train.csv') self.assertEqual(100, len(data)) + def test_plot(self): + # This relies on the hvplot package + data = pl.read_csv('/input/tests/data/train.csv') + data.plot.line() +