Merge pull request #356 from boazmohar/block_chunk_size

Bug fix for images.toblocks()
thunder-project · Aug 19, 2016 · c6054a4 · c6054a4
2 parents e795388 + b410fbb
commit c6054a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thunder/images/images.py b/thunder/images/images.py
@@ -78,7 +78,7 @@ def toblocks(self, chunk_size='auto', padding=None):
 
         if self.mode == 'spark':
             if chunk_size is 'auto':
-                chunk_size = str(int(100000.0/self.shape[0]))
+                chunk_size = str(max([int(1e5/self.shape[0]), 1]))
             chunks = self.values.chunk(chunk_size, padding=padding).keys_to_values((0,))
 
         if self.mode == 'local':