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