little fix

matsengrp · Nov 20, 2024 · b8a0925 · b8a0925
1 parent b2601a4
commit b8a0925
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/netam/dxsm.py b/netam/dxsm.py
@@ -73,7 +73,7 @@ def __init__(
 
         for i, (aa_parent, aa_child) in enumerate(zip(aa_parents, aa_children)):
             self.masks[i, :] = codon_mask_tensor_of(
-                nt_parents[i], nt_children[i], self.max_aa_seq_len
+                nt_parents[i], nt_children[i], aa_length=self.max_aa_seq_len
             )
             aa_seq_len = len(aa_parent)
             assert_pcp_valid(