From de2c68fa48524c37a59e576e0352788941ed9c36 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Thu, 19 Dec 2019 21:24:23 +0100
Subject: fix class

---
 inversion/live.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inversion/live.py b/inversion/live.py
index 6894b47..b5ac0c3 100644
--- a/inversion/live.py
+++ b/inversion/live.py
@@ -67,7 +67,7 @@ def clamp(n, a=0, b=1):
 def label_sampler(num_classes=1, shape=(BATCH_SIZE, N_CLASS,)):
   label = np.zeros(shape)
   for i in range(shape[0]):
-    for _ in range(num_classes):
+    for _ in range(int(num_classes)):
       j = random.randint(0, shape[1]-1)
       label[i, j] = random.random()
       print("class: {} {}".format(j, label[i, j]))
-- 
cgit v1.2.3-70-g09d2