From 6fc601ae4e2d25b555549aee6c4675e4f7324be5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=BCdiger=20Busche?= Date: Fri, 16 Jul 2021 10:27:12 +0200 Subject: [PATCH] Fix pickle bug with pandas 1.3.0 (#38) Co-authored-by: rbusche --- requirements.txt | 2 +- tests/conftest.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 9d1e01a..fe9b12c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ numpy matplotlib -pandas +pandas==1.3.0 plotnine scipy seaborn diff --git a/tests/conftest.py b/tests/conftest.py index 32909be..d77dde0 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -42,8 +42,7 @@ def tsc_generator(train_data, test_data): @pytest.fixture def filter_combination(shared_datadir): - with open(shared_datadir / "cases.pickle", "rb") as handle: - cases = pickle.load(handle) + cases = pd.read_pickle(shared_datadir / "cases.pickle") cases_in_berlin = cases.query('county == "Berlin"') return FilterCombination( disease="SAL", county="Berlin", pathogen="SAL", data=cases_in_berlin