This seems to help

mila-iqia · Jan 17, 2015 · ab85ad0 · ab85ad0
1 parent 5c02402
commit ab85ad0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/blocks/datasets/schemes.py b/blocks/datasets/schemes.py
@@ -31,6 +31,7 @@ def get_request_iterator(self):
         raise NotImplementedError
 
 
+@add_metaclass(ABCMeta)
 class BatchSizeScheme(IterationScheme):
     """Iteration scheme that returns batch sizes.
 
@@ -43,6 +44,7 @@ class BatchSizeScheme(IterationScheme):
     pass
 
 
+@add_metaclass(ABCMeta)
 class BatchScheme(IterationScheme):
     """Iteration schemes that return slices or indices for batches.