fairseq/modules/kmeans_vector_quantizer.py

Killed 0 out of 3 mutants

Survived

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

Mutant 1915

--- fairseq/modules/kmeans_vector_quantizer.py
+++ fairseq/modules/kmeans_vector_quantizer.py
@@ -11,7 +11,7 @@
 
 class KmeansVectorQuantizer(nn.Module):
     def __init__(
-        self, dim, num_vars, groups, combine_groups, vq_dim, time_first, gamma=0.25
+        self, dim, num_vars, groups, combine_groups, vq_dim, time_first, gamma=1.25
     ):
         '''Vector quantization using straight pass-through estimator (i.e. kmeans)

Mutant 1916

--- fairseq/modules/kmeans_vector_quantizer.py
+++ fairseq/modules/kmeans_vector_quantizer.py
@@ -60,7 +60,6 @@
 
         return y.detach() + (x - x.detach())
 
-    @property
     def expand_embedding(self):
         if self.combine_groups:
             return self.embedding.expand(self.num_vars, self.groups, self.var_dim)

Mutant 1917

--- fairseq/modules/kmeans_vector_quantizer.py
+++ fairseq/modules/kmeans_vector_quantizer.py
@@ -70,7 +70,7 @@
         res = self.forward(x, produce_targets=True)
         return res["x"], res["targets"]
 
-    def forward(self, x, produce_targets=False):
+    def forward(self, x, produce_targets=True):
 
         result = {"num_vars": self.num_vars}