1 files changed, 3 insertions, 1 deletions
diff --git a/model.py b/model.py
index ab438c7..1d2a74d 100644
--- a/model.py
+++ b/model.py
@@ -340,16 +340,18 @@ class PrimedGenerator(Runner):
         def noise(x):
             for i in xrange(n_samples):
                 x[:, i] = random.triangular(q_min, q_max)
+            return x
         def sin(x):
             primer_freq = prime_param_a
             for i in xrange(n_samples):
                 x[:, i] = (math.sin(i/44100 * primer_freq) + 1) / 2 * (q_max - q_min) + q_min
+            return x
 
         sequences = {
           'zero': lambda x: x.fill_(utils.q_zero(self.model.q_levels)),
           'noise': noise,
           'sin': sin,
-        }.get(value, default)(sequences)
+        }.get(primer, 'zero')(sequences)
 
         for i in range(self.model.lookback, self.model.lookback + seq_len):
             for (tier_index, rnn) in \