missed bits

matsengrp · Nov 19, 2024 · 2e62463 · 2e62463
1 parent 776ab82
commit 2e62463
Show file tree

Hide file tree

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