# HG changeset patch
# User Thierry Bertin-Mahieux <bertinmt@iro.umontreal.ca>
# Date 1213647096 14400
# Node ID 20e08c52c98c8102c8a4931f8eeb9e4bf713c790
# Parent  9ce791fb2cbf1721ce66d01610a7a8bebf52db50# Parent  09140ba68e17501e9f676db12b9ddf0fbaac2e08
merge

diff -r 09140ba68e17 -r 20e08c52c98c dataset.py
--- a/dataset.py	Mon Jun 16 16:06:59 2008 -0400
+++ b/dataset.py	Mon Jun 16 16:11:36 2008 -0400
@@ -240,7 +240,13 @@
         def next(self):
             size1_minibatch = self.minibatch_iterator.next()
             if not self.minibatch:
-                self.minibatch = Example(size1_minibatch.keys(),[value[0] for value in size1_minibatch.values()])
+                names = size1_minibatch.keys()
+                # next lines are a hack, but there was problem when we were getting [array(327)] for instance
+                if len(size1_minibatch.values()[0].shape) > 0 :
+                    values = [value[0] for value in size1_minibatch.values()]
+                else :
+                    values = [value for value in size1_minibatch.values()]
+                self.minibatch = Example(names,values)
             else:
                 self.minibatch._values = [value[0] for value in size1_minibatch.values()]
             return self.minibatch