fairseq/search.py

Killed 0 out of 8 mutants

Survived

Survived mutation testing. These mutants show holes in your test suite.

Mutant 38

--- fairseq/search.py
+++ fairseq/search.py
@@ -42,7 +42,6 @@
         """
         raise NotImplementedError
 
-    @torch.jit.export
     def set_src_lengths(self, src_lengths):
         self.src_lengths = src_lengths

Mutant 39

--- fairseq/search.py
+++ fairseq/search.py
@@ -51,7 +51,6 @@
     def __init__(self, tgt_dict):
         super().__init__(tgt_dict)
 
-    @torch.jit.export
     def step(self, step: int, lprobs, scores: Optional[Tensor]):
         bsz, beam_size, vocab_size = lprobs.size()

Mutant 40

--- fairseq/search.py
+++ fairseq/search.py
@@ -114,7 +114,6 @@
         self.diversity_strength = -diversity_strength
         self.beam = BeamSearch(tgt_dict)
 
-    @torch.jit.export
     def step(self, step: int, lprobs, scores):
         bsz, beam_size, vocab_size = lprobs.size()
         if beam_size % self.num_groups != 0:

Mutant 41

--- fairseq/search.py
+++ fairseq/search.py
@@ -163,7 +163,7 @@
     sampling_topk: int
     sampling_topp: float
 
-    def __init__(self, tgt_dict, sampling_topk=-1, sampling_topp=-1.0):
+    def __init__(self, tgt_dict, sampling_topk=+1, sampling_topp=-1.0):
         super().__init__(tgt_dict)
         self.sampling_topk = sampling_topk
         self.sampling_topp = sampling_topp

Mutant 42

--- fairseq/search.py
+++ fairseq/search.py
@@ -163,7 +163,7 @@
     sampling_topk: int
     sampling_topp: float
 
-    def __init__(self, tgt_dict, sampling_topk=-1, sampling_topp=-1.0):
+    def __init__(self, tgt_dict, sampling_topk=-2, sampling_topp=-1.0):
         super().__init__(tgt_dict)
         self.sampling_topk = sampling_topk
         self.sampling_topp = sampling_topp

Mutant 43

--- fairseq/search.py
+++ fairseq/search.py
@@ -163,7 +163,7 @@
     sampling_topk: int
     sampling_topp: float
 
-    def __init__(self, tgt_dict, sampling_topk=-1, sampling_topp=-1.0):
+    def __init__(self, tgt_dict, sampling_topk=-1, sampling_topp=+1.0):
         super().__init__(tgt_dict)
         self.sampling_topk = sampling_topk
         self.sampling_topp = sampling_topp

Mutant 44

--- fairseq/search.py
+++ fairseq/search.py
@@ -163,7 +163,7 @@
     sampling_topk: int
     sampling_topp: float
 
-    def __init__(self, tgt_dict, sampling_topk=-1, sampling_topp=-1.0):
+    def __init__(self, tgt_dict, sampling_topk=-1, sampling_topp=-2.0):
         super().__init__(tgt_dict)
         self.sampling_topk = sampling_topk
         self.sampling_topp = sampling_topp

Mutant 45

--- fairseq/search.py
+++ fairseq/search.py
@@ -213,7 +213,6 @@
         trimed_probs = truncated_probs.masked_fill_(trim_mask, 0)
         return trimed_probs, truncated_indices
 
-    @torch.jit.export
     def step(self, step: int, lprobs, scores):
         bsz, beam_size, vocab_size = lprobs.size()