diff --git a/examples/DENV2/py/resample_aln.py b/examples/DENV2/py/resample_aln.py index 79b7acf..c8ab78b 100644 --- a/examples/DENV2/py/resample_aln.py +++ b/examples/DENV2/py/resample_aln.py @@ -27,7 +27,7 @@ used_ids = Counter() df = pd.read_csv(params.input_tab, header=0, index_col=0, sep='\t') - df = df.loc[set(ids), :] + df = df.loc[list(set(ids)), :] def get_seq(id): seq = id2seq[id] diff --git a/pastml/acr.py b/pastml/acr.py index 1c8dcd9..ab8b877 100755 --- a/pastml/acr.py +++ b/pastml/acr.py @@ -862,7 +862,7 @@ def _validate_input(tree_nwk, columns=None, name_column=None, data=None, data_se for n in root.traverse(): n.name = n.name.strip("'").strip('"') - filtered_df = df.loc[common_ids, :] + filtered_df = df.loc[list(common_ids), :] if not filtered_df.shape[0]: tip_name_representatives = [] for _ in roots[0].iter_leaves():