style: auto-fixes from pre-commit

Aarhus-Psychiatry-Research · Nov 15, 2023 · 8db97a8 · 8db97a8
1 parent a54d08c
commit 8db97a8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/psycop/common/model_training_v2/trainer/preprocessing/pipeline.py b/psycop/common/model_training_v2/trainer/preprocessing/pipeline.py
@@ -21,7 +21,7 @@ def apply(self, data: PolarsFrame) -> pd.DataFrame:
 
 
 @BaselineRegistry.preprocessing.register("baseline_preprocessing_pipeline")
-class BaselinePreprocessingPipeline(PreprocessingPipeline): 
+class BaselinePreprocessingPipeline(PreprocessingPipeline):
     def __init__(self, *args: PresplitStep) -> None:
         self.steps = list(args)
 

diff --git a/psycop/common/model_training_v2/trainer/preprocessing/steps/row_filters.py b/psycop/common/model_training_v2/trainer/preprocessing/steps/row_filters.py
@@ -9,7 +9,12 @@
 
 @BaselineRegistry.preprocessing.register("age_filter")
 class AgeFilter(PresplitStep):
-    def __init__(self, min_age: int, max_age: int = 999, age_col_name: str = "pred_age"):
+    def __init__(
+        self,
+        min_age: int,
+        max_age: int = 999,
+        age_col_name: str = "pred_age",
+    ):
         self.min_age = min_age
         self.max_age = max_age
         self.age = pl.col(age_col_name)