Can we add an option to `partial_map` and `map`?