Skip to content

Commit d8e5a1e

Browse files
Add wcvp version to generic_prepare_data
1 parent 54f5983 commit d8e5a1e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

data_compilation_methods/preparing_datasets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def get_tempout_csv(dataset_name: str, temp_output_path: str):
1414
def generic_prepare_data(dataset_name: str, output_csv: str, df: pd.DataFrame, name_col: str,
1515
snippet_column: str,
1616
dropifna: List[str] = None, families_of_interest: List[str] = None,
17-
batch: bool = False, family_column: str = None):
17+
batch: bool = False, family_column: str = None, wcvp_version: str = None):
1818
from wcvpy.wcvp_name_matching import get_accepted_info_from_names_in_column
1919

2020
if dropifna is not None:
@@ -34,10 +34,10 @@ def generic_prepare_data(dataset_name: str, output_csv: str, df: pd.DataFrame, n
3434
acc_dfs.append(
3535
get_accepted_info_from_names_in_column(split_df, name_col,
3636
families_of_interest=families_of_interest,
37-
family_column=family_column))
37+
family_column=family_column, wcvp_version=wcvp_version))
3838
db_acc = pd.concat(acc_dfs)
3939
else:
4040
db_acc = get_accepted_info_from_names_in_column(df, name_col,
4141
families_of_interest=families_of_interest,
42-
family_column=family_column)
42+
family_column=family_column, wcvp_version=wcvp_version)
4343
db_acc.to_csv(output_csv)

0 commit comments

Comments
 (0)