Skip to content
This repository has been archived by the owner on May 3, 2023. It is now read-only.

Commit

Permalink
fix: add skema_2_without_nutrition again
Browse files Browse the repository at this point in the history
  • Loading branch information
signekb committed Mar 14, 2023
1 parent 505ea4d commit 685c5cb
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions src/psycop_feature_generation/loaders/raw/load_coercion.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 685c5cb

Please sign in to comment.