Merge pull request #201 from Aarhus-Psychiatry-Research/skema_2_witho…

…ut_nutrition fix: add skema_2_without_nutrition again
Aarhus-Psychiatry-Research · Mar 14, 2023 · 8dcf0df · 8dcf0df · github-actions · Mar 14, 2023
2 parents 505ea4d + 685c5cb
commit 8dcf0df
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/psycop_feature_generation/loaders/raw/load_coercion.py b/src/psycop_feature_generation/loaders/raw/load_coercion.py
@@ -278,6 +278,30 @@ def skema_2(
     )
 
 
+@data_loaders.register("skema_2_without_nutrition")
+def skema_2_without_nutrition(
+    n_rows: Optional[int] = None,
+    unpack_to_intervals: Optional[bool] = False,
+    unpack_freq: Optional[str] = "D",
+) -> pd.DataFrame:
+    coercion_types_list = [
+        {
+            "coercion_type": "Af legemlig lidelse",
+        },
+        {
+            "coercion_type": "Medicinering",
+        },
+        {
+            "coercion_type": "ECT",
+        },
+    ]
+
+    return _concatenate_coercion(
+        coercion_types_list=coercion_types_list,
+        n_rows=n_rows,
+    )
+
+
 # magtanvendelse
 @data_loaders.register("skema_3")
 def skema_3(