fixed some typing pyright errors

alexliap · Mar 8, 2024 · b154f18 · b154f18
1 parent 6ef5735
commit b154f18
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/roll_rate_analysis/mom_roll_rate_table.py b/src/roll_rate_analysis/mom_roll_rate_table.py
@@ -101,7 +101,11 @@ def build(self):
         )
 
     def _get_temp_data(
-        self, data: pl.DataFrame, case: int, cycle: int = None, priority: int = None
+        self,
+        data: pl.DataFrame,
+        case: int,
+        cycle: int | None = None,
+        priority: int | None = None,
     ):
         """
         Get a temporary grouped part of the data, for every step of the roll rate calculation procedure.

diff --git a/src/roll_rate_analysis/snapshot_roll_rate_table.py b/src/roll_rate_analysis/snapshot_roll_rate_table.py
@@ -52,7 +52,7 @@ def __init__(
         delinquency_col: str,
         obs_files: list[str],
         perf_files: list[str],
-        keep_cols: list[str] = None,
+        keep_cols: list[str] | None = None,
         max_delq: int = 6,
         detailed: bool = False,
         granularity: int = 1,