From 6a9cc3f30d1d12f6cf43c56f6d60f285c4158e9a Mon Sep 17 00:00:00 2001 From: Dani Arribas-Bel Date: Fri, 18 Aug 2023 14:04:46 +0100 Subject: [PATCH] Minor fix for dask test --- tobler/tests/test_area_interpolators.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tobler/tests/test_area_interpolators.py b/tobler/tests/test_area_interpolators.py index d5adf61..9070aa9 100644 --- a/tobler/tests/test_area_interpolators.py +++ b/tobler/tests/test_area_interpolators.py @@ -93,12 +93,9 @@ def test_area_interpolate_categorical_dask(): .spatial_shuffle(by='hilbert', shuffle='tasks') ) area = area_interpolate_dask( - source_df=sac1, - target_df=sac2, - extensive_variables=["TOT_POP"], - intensive_variables=["pct_poverty"], + source_dgdf=sac1, + target_dgdf=sac2, categorical_variables=["animal"], - n_jobs=1, ) assert_almost_equal(area.animal_cat.sum(), 32, decimal=0) assert_almost_equal(area.animal_dog.sum(), 19, decimal=0)