Fix pickle for Dataset objects

Fixes #167

Related: #166 (though I'm not sure why that didn't work)
Merge pull request #168 from shoyer/fix-pickle

Fix pickle for Dataset objects