fix types

1 files changed, 3 insertions, 3 deletions

diff --git a/model.py b/model.py
index 6d221ce..d844743 100644
--- a/model.py
+++ b/model.py
@@ -341,12 +341,12 @@ class PrimedGenerator(Runner):
 
         def _noise(x):
             for i in range(n_samples):
-                x[:, i] = random.triangular(q_min, q_max)
+                x[:, i] = int(random.triangular(q_min, q_max))
             return x
         def _sin(x):
-            primer_freq = prime_param_a
+            primer_freq = float(prime_param_a)
             for i in range(n_samples):
-                x[:, i] = (math.sin(i/44100 * primer_freq) + 1) / 2 * (q_max - q_min) + q_min
+                x[:, i] = int((math.sin(i/44100 * primer_freq) + 1) / 2 * (q_max - q_min) + q_min)
             return x
 
         sequences = {