From 17841d328ff99ae7a49b8228c354262584b98c78 Mon Sep 17 00:00:00 2001 From: Dani Arribas-Bel Date: Fri, 18 Aug 2023 14:45:01 +0100 Subject: [PATCH] loading dask gdf to memory for tests --- tobler/tests/test_area_interpolators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tobler/tests/test_area_interpolators.py b/tobler/tests/test_area_interpolators.py index 4531ea3..bc8791b 100644 --- a/tobler/tests/test_area_interpolators.py +++ b/tobler/tests/test_area_interpolators.py @@ -97,7 +97,7 @@ def test_area_interpolate_categorical_dask(): target_dgdf=dsac2, id_col='ZIP', categorical_variables=["animal"], - ) + ).compute() assert_almost_equal(area.animal_cat.sum(), 32, decimal=0) assert_almost_equal(area.animal_dog.sum(), 19, decimal=0) assert_almost_equal(area.animal_donkey.sum(), 22, decimal=0)