I don't think we need to use `assert_valid_keys` if users provide more than default it should just work. No need to raise exception.