FIX set_params needs to return ``self``