Skip to content

Commit

Permalink
Fixing some warnings
Browse files Browse the repository at this point in the history
  • Loading branch information
matsen committed Sep 27, 2024
1 parent 2b1114b commit 6b2f19c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_molevol.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,11 +149,11 @@ def test_aaprob_of_mut_and_sub():
crepe_path = "data/cnn_joi_sml-shmoof_small"
crepe = framework.load_crepe(crepe_path)
[rates], [subs] = crepe([parent_nt_seq])
mut_probs = 1.0 - torch.exp(-torch.tensor(rates.squeeze()))
mut_probs = 1.0 - torch.exp(-rates.squeeze().clone().detach())
parent_codon = parent_nt_seq[0:3]
parent_codon_idxs = nt_idx_tensor_of_str(parent_codon)
codon_mut_probs = mut_probs[0:3]
codon_subs = torch.tensor(subs[0:3])
codon_subs = subs.clone().detach()[0:3]

iterative_result = iterative_aaprob_of_mut_and_sub(
parent_codon, codon_mut_probs, codon_subs
Expand Down

0 comments on commit 6b2f19c

Please sign in to comment.