Speedup read_excel for SMHI data Klimatbyran#682

This shaved off 6 seconds from the test. * Use calamine when importing SMHI data
joakimbits · Sep 19, 2024 · 216339e · 216339e
1 parent 28138b8
commit 216339e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/data/issues/emissions/historical_data_calculations.py b/data/issues/emissions/historical_data_calculations.py
@@ -12,7 +12,7 @@ def get_smhi_data():
         pandas.DataFrame: The dataframe containing the SMHI data.
     """
 
-    df_raw = pd.read_excel(PATH_SMHI)
+    df_raw = pd.read_excel(PATH_SMHI, engine="calamine")
 
     # Remove the first 4 rows and reset the index
     df_raw = df_raw.drop([0, 1, 2]).reset_index(drop=True)

diff --git a/data/requirements.txt b/data/requirements.txt
@@ -7,3 +7,4 @@ scipy
 pytest
 pytest-profiling
 graphviz
+python-calamine