wrappers: explicitly case discrete states to unicode to avoid str~flo…

…at comparison when pandas parses missing data as np.nan
neherlab · Oct 18, 2019 · 24a6f4a · 24a6f4a
1 parent d1b8cab
commit 24a6f4a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/treetime/wrappers.py b/treetime/wrappers.py
@@ -754,7 +754,7 @@ def mugration(params):
         attr = states.columns[1]
         print("Attribute for mugration inference was not specified. Using "+attr, file=sys.stderr)
 
-    leaf_to_attr = {x[taxon_name]:x[attr] for xi, x in states.iterrows()
+    leaf_to_attr = {x[taxon_name]:str(x[attr]) for xi, x in states.iterrows()
                     if x[attr]!=params.missing_data}
 
     mug, letter_to_state, reverse_alphabet = reconstruct_discrete_traits(params.tree, leaf_to_attr, missing_data=params.missing_data,