It looks like this will require passing a keyword argument to method.py#fit.

I think there also needs to be some specification of the random state to make sure the train/test split is preserved on repeated calls.