I'm looking through the model selection code and I find myself curious what the various inputs and outputs are to functions. This might be a bit of our codebase that could use with some mild type annotations.

This might be a good activity for someone who wanted to become familiar with the Dask-ML codebase.