diff --git a/netam/dxsm.py b/netam/dxsm.py index 8bda5346..60642df3 100644 --- a/netam/dxsm.py +++ b/netam/dxsm.py @@ -56,7 +56,6 @@ def __init__( branch_lengths: torch.Tensor, multihit_model=None, ): - print("starting DXSMDataset init") self.nt_parents = nt_parents self.nt_children = nt_children self.nt_ratess = nt_ratess @@ -79,7 +78,6 @@ def __init__( aa_parents = translate_sequences(self.nt_parents) aa_children = translate_sequences(self.nt_children) self.max_aa_seq_len = max(len(seq) for seq in aa_parents) - print("max_aa_seq_len:", self.max_aa_seq_len) # We have sequences of varying length, so we start with all tensors set # to the ambiguous amino acid, and then will fill in the actual values # below. diff --git a/tests/test_ambiguous.py b/tests/test_ambiguous.py index 3f996835..845a6c26 100644 --- a/tests/test_ambiguous.py +++ b/tests/test_ambiguous.py @@ -149,6 +149,7 @@ def test_dnsm_burrito(ambig_pcp_df, dnsm_model): burrito.joint_train(epochs=1, cycle_count=2, training_method="full") return burrito + @pytest.fixture def dasm_model(): return TransformerBinarySelectionModelWiggleAct(