From 6d9d6f2c92705902b3d2166cec9f5681e10eab4d Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 14 May 2018 20:46:55 +0200
Subject: okkkkkkkk

---
 model.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'model.py')

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 \
-- 
cgit v1.2.3-70-g09d2