From f97aee8ff932270abed737308591cc87678062a8 Mon Sep 17 00:00:00 2001 From: bokajgd Date: Tue, 18 Oct 2022 11:10:41 +0200 Subject: [PATCH] fix: remove shak_code + operator check Fixes #70 --- src/psycop_feature_generation/loaders/raw/load_admissions.py | 5 ----- src/psycop_feature_generation/loaders/raw/load_visits.py | 5 ----- 2 files changed, 10 deletions(-) diff --git a/src/psycop_feature_generation/loaders/raw/load_admissions.py b/src/psycop_feature_generation/loaders/raw/load_admissions.py index 20b381e9..a95f7d58 100644 --- a/src/psycop_feature_generation/loaders/raw/load_admissions.py +++ b/src/psycop_feature_generation/loaders/raw/load_admissions.py @@ -27,11 +27,6 @@ def admissions( pd.DataFrame: Dataframe with all physical visits to psychiatry. Has columns dw_ek_borger, timestamp and value (length of admissions in days). """ - if bool(shak_code) != bool(shak_sql_operator): - raise ValueError( - "shak_code and shak_sql_operator must both be set or both be None.", - ) - # SHAK = 6600 ≈ in psychiatry d = { "LPR3": { diff --git a/src/psycop_feature_generation/loaders/raw/load_visits.py b/src/psycop_feature_generation/loaders/raw/load_visits.py index 0cbc7508..3d9797a1 100644 --- a/src/psycop_feature_generation/loaders/raw/load_visits.py +++ b/src/psycop_feature_generation/loaders/raw/load_visits.py @@ -31,11 +31,6 @@ def physical_visits( pd.DataFrame: Dataframe with all physical visits to psychiatry. Has columns dw_ek_borger and timestamp. """ - if bool(shak_code) != bool(shak_sql_operator): - raise ValueError( - "shak_code and shak_sql_operator must both be set or both be None.", - ) - # SHAK = 6600 ≈ in psychiatry d = { "LPR3": {