Can you wrap the methods like this https://github.com/lisa-lab/pylearn2/blob/master/pylearn2/models/mlp.py#L410.
Do this for all methods inherited from Datasets class. It helps for the doc.