Add with_split to DatasetDict.map

[jp1924]

#7356

[jp1924]

Can you check this out, @lhoestq?

[jp1924]

cc @lhoestq @albertvillanova

[jp1924]

@lhoestq

[jp1924]

@lhoestq

[jp1924]

@lhoestq

[lhoestq]

to avoid this issue we use positional arguments for indices/rank to avoid this issue (this way users can name their parameters arbitrarily), maybe we can use this trick: https://stackoverflow.com/a/66274908 ?

[jp1924]

@lhoestq
Ah, I see what you mean.

So what you're saying is that there may be people who were already using fn_kwagrs to do splits, and my implementation may cause conflicts? I agree.

So let's use partial instead of fn_kwargs to solve the problem? Sounds like a good idea!

So I redefined partial and made it like this, can you take a look?

class bind(partial):
    def __call__(self, *fn_args, **fn_kwargs):
        return self.func(*fn_args, *self.args, **fn_kwargs)