Fix pickle bug with pandas 1.3.0 (#38)

Co-authored-by: rbusche <[email protected]>
JarnoRFB · Jul 16, 2021 · 6fc601a · 6fc601a
1 parent ed297b4
commit 6fc601a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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