fixed pandas compatibility

evolbioinfo · Jul 10, 2023 · e95c0e9 · e95c0e9
1 parent 985867d
commit e95c0e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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():