I think you should explicitly fix the number of samples and features here to make sure that the test run quickly enough (ideally under 1s) even if the default arguments of `datasets.make_multilabel_classification` change in the future.