I think this will pass through a `sample_weight` that is a list, which will break on `sample_weight[train]`, etc. Could you please make every test with `sample_weight` ensure that a list can be used? You don't seem to have any tests for this module yet.