Skip to content

Commit 4d20d3c

Browse files
committed
add typing hint
1 parent 2a8b7b8 commit 4d20d3c

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

genie/database_to_staging.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,9 @@ def _redact_ped_year(df_col: pd.Series) -> pd.Series:
166166

167167

168168
# TODO: Add to transform.py
169-
def _to_redact_difference(df_col_year1: pd.Series, df_col_year2: pd.Series) -> pd.Series:
169+
def _to_redact_difference(
170+
df_col_year1: pd.Series, df_col_year2: pd.Series
171+
) -> pd.Series:
170172
"""Determine if difference between year2 and year1 is > 89
171173
172174
Args:
@@ -186,7 +188,8 @@ def _to_redact_difference(df_col_year1: pd.Series, df_col_year2: pd.Series) -> p
186188

187189
# TODO: Add to transform.py
188190
def redact_phi(
189-
clinicaldf: pd.DataFrame, interval_cols_to_redact=["AGE_AT_SEQ_REPORT", "INT_CONTACT", "INT_DOD"]: list
191+
clinicaldf: pd.DataFrame,
192+
interval_cols_to_redact: list = ["AGE_AT_SEQ_REPORT", "INT_CONTACT", "INT_DOD"],
190193
) -> pd.DataFrame:
191194
"""Redacts the PHI by re-annotating the clinical file
192195

0 commit comments

Comments
 (0)