Merge pull request #47 from Aarhus-Psychiatry-Research/bokajgd/issue38

Bokajgd/issue38
Aarhus-Psychiatry-Research · Oct 10, 2022 · c58fcfc · c58fcfc
2 parents 2eee15c + ea0d946
commit c58fcfc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/psycop_feature_generation/data_checks/flattened/data_integrity.py b/src/psycop_feature_generation/data_checks/flattened/data_integrity.py
@@ -171,6 +171,7 @@ def check_train_data_integrity(
         df=train_predictors,
         index_name="dw_ek_borger",
         datetime_name="timestamp",
+        cat_features=[],
     )
 
     # Running checks that do not require a label
@@ -194,6 +195,7 @@ def check_train_data_integrity(
             index_name="dw_ek_borger",
             datetime_name="timestamp",
             label=train_outcomes_df[outcome_column],
+            cat_features=[],
         )
 
         suite_results = label_checks.run(data_s)
@@ -281,6 +283,7 @@ def get_split_as_ds_dict(
         df=predictors,
         index_name="dw_ek_borger",
         datetime_name="timestamp",
+        cat_features=[],
     )
 
     return {