Skip to content

Commit

Permalink
Fix type annotations
Browse files Browse the repository at this point in the history
  • Loading branch information
otaviocv committed Aug 15, 2023
1 parent b7b7229 commit ba5e957
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/fklearn/training/transformation.py
Original file line number Diff line number Diff line change
Expand Up @@ -1030,7 +1030,7 @@ def p(dataset: pd.DataFrame) -> pd.DataFrame:
def detailed_assignment(df: pd.DataFrame, cols_to_check: List[str]) -> list:
print(df.loc[:, cols_to_check])
cols_with_missing = np.array([np.where(df[col].isna(), col, "") for col in cols_to_check]).T
missing_by_row_list = [list(filter(None, x)) for x in cols_with_missing]
missing_by_row_list: list = [list(filter(None, x)) for x in cols_with_missing]
if len(missing_by_row_list) == 0:
return np.empty((0, 0)).tolist()
else:
Expand Down

0 comments on commit ba5e957

Please sign in to comment.