From 0e52813a4e435cddeb80ec9972b2bf1fc791c0cc Mon Sep 17 00:00:00 2001
From: Alex Auvolat <alex@adnab.me>
Date: Sun, 21 Jun 2015 14:33:47 -0400
Subject: Updates for Blocks compatibility

---
 data/hdf5.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'data')

diff --git a/data/hdf5.py b/data/hdf5.py
index 6e2f9a4..f3d6da2 100644
--- a/data/hdf5.py
+++ b/data/hdf5.py
@@ -13,7 +13,7 @@ class TaxiDataset(H5PYDataset):
     def __init__(self, which_set, filename='data.hdf5', **kwargs):
         self.filename = filename
         kwargs.setdefault('load_in_memory', True)
-        super(TaxiDataset, self).__init__(self.data_path, which_set, **kwargs)
+        super(TaxiDataset, self).__init__(self.data_path, (which_set,), **kwargs)
 
     @property
     def data_path(self):
-- 
cgit v1.2.3