Add seaborn theme initialization

src/analysis/main.py

@@ -94,7 +94,7 @@ def main(args: argparse.Namespace):
     df: pd.DataFrame | DataFrameGroupBy = date_filter(df, start_date, end_date)
     logging.info(f"Loaded {len(df)} data points ({original_length} before filtering)")
 
-    stat.set_plot_title(df, args)
+    stat.prepare_mpl(df, args)
 
     if args.group_by != "none":
         df_grouped: DataFrameGroupBy | None = None

src/analysis/stat.py

 import argparse
 
+import matplotlib as mpl
 import matplotlib.pyplot as plt
 import pandas as pd
 import seaborn as sns
@@ -13,14 +14,17 @@ def describe(df: pd.DataFrame | DataFrameGroupBy) -> None:
     print(df.describe())
 
 
-def set_plot_title(df: pd.DataFrame, args: argparse.Namespace) -> None:
-    """Set the plot title based on the cli arguments"""
+def prepare_mpl(df: pd.DataFrame, args: argparse.Namespace) -> None:
+    """Prepare matplotlib params"""
     if args.stat not in [
         "delay_boxplot",
         "day_train_count",
     ]:
         return
 
+    mpl.rcParams["figure.figsize"] = (12, 12 * 5 / 7)
+    sns.set_theme(style="whitegrid", palette="pastel")
+
     plt.title(args.stat)
 
     start_day, end_day = df.day.min().date(), df.day.max().date()
@@ -79,7 +83,6 @@ def delay_boxplot(df: pd.DataFrame | DataFrameGroupBy) -> None:
         )
         ax.set(xlabel="Variable", ylabel="Delay (minutes)")
 
-    plt.grid()
     plt.show()