Merge pull request #134 from daikitag/genetic

CODE: Genetic value
tskit-dev · Feb 12, 2024 · 19454d6 · 19454d6
2 parents 34dcec4 + 7a16cce
commit 19454d6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tstrait/genetic_value.py b/tstrait/genetic_value.py
@@ -198,6 +198,8 @@ def genetic_value(ts, trait_df):
     if np.min(trait_id) != 0 or np.max(trait_id) != len(trait_id) - 1:
         raise ValueError("trait_id must be consecutive and start from 0")
 
+    trait_df = trait_df.astype({"trait_id": int})
+
     genetic = _GeneticValue(ts=ts, trait_df=trait_df)
 
     genetic_result = genetic._run()