for future reference, sphinx actually wants there to be a space between the parameter name and the colon:

    which_set : str

but I won't let that block this PR