needleman_wunsch

Genentech · Apr 29, 2024 · 72c3287 · 72c3287
1 parent c174a4e
commit 72c3287
Showing 1 changed file with 0 additions and 15 deletions.
diff --git a/src/beignet/operators/_needleman_wunsch.py b/src/beignet/operators/_needleman_wunsch.py
@@ -1,8 +1,6 @@
 import math
 import operator
 
-import matplotlib.pyplot
-import numpy.random
 import torch
 import torch.func
 import torch.nn.functional
@@ -194,16 +192,3 @@ def fn(x: Tensor, shape: Tensor) -> Tensor:
         output[index] = torch.func.grad(fn)(input[index], lengths[index])
 
     return output
-
-
-def test():
-    x = torch.tensor(numpy.random.random((32, 8, 8)), device="cpu")
-
-    lengths = torch.full([32, 2], 8, device="cpu")
-
-    alignment = needleman_wunsch(x, lengths, -1.0, 1.0)[0]
-
-    matplotlib.pyplot.imshow(alignment.cpu().numpy())
-
-
-test()