Access to key in DatasetDict map
MEMBER
Indeed that would be cool

Also FYI right now the easiest way to do this is
```python
dataset_dict["train"] = dataset_dict["train"].map(my_transform_for_the_train_set)
dataset_dict["test"] = dataset_dict["test"].map(my_transform_for_the_test_set)
```
2021-01-13T10:37:51Z
https://github.com/huggingface/datasets/issues/1593#issuecomment-759361862
https://api.github.com/repos/huggingface/datasets/issues/1593
MDEyOklzc3VlQ29tbWVudDc1OTM2MTg2Mg==
https://api.github.com/repos/huggingface/datasets/issues/comments/759361862/reactions
2021-01-13T10:38:03Z
https://api.github.com/repos/huggingface/datasets/issues/comments/759361862
https://avatars.githubusercontent.com/u/42851186?v=4
https://api.github.com/users/lhoestq/events{/privacy}
https://api.github.com/users/lhoestq/followers
https://api.github.com/users/lhoestq/following{/other_user}
https://api.github.com/users/lhoestq/gists{/gist_id}
https://github.com/lhoestq
lhoestq
MDQ6VXNlcjQyODUxMTg2
https://api.github.com/users/lhoestq/orgs
https://api.github.com/users/lhoestq/received_events
https://api.github.com/users/lhoestq/repos
https://api.github.com/users/lhoestq/starred{/owner}{/repo}
https://api.github.com/users/lhoestq/subscriptions
User
https://api.github.com/users/lhoestq
public
COLLABORATOR
I don't feel like adding an extra param for this simple usage makes sense, considering how many args `map` already has. 

(Feel free to re-open this issue if you don't agree with me)
2022-10-05T12:33:06Z
https://github.com/huggingface/datasets/issues/1593#issuecomment-1268375446
https://api.github.com/repos/huggingface/datasets/issues/1593
IC_kwDODunzps5Lmd-W
https://api.github.com/repos/huggingface/datasets/issues/comments/1268375446/reactions
2022-10-05T12:33:06Z
https://api.github.com/repos/huggingface/datasets/issues/comments/1268375446
https://avatars.githubusercontent.com/u/47462742?v=4
https://api.github.com/users/mariosasko/events{/privacy}
https://api.github.com/users/mariosasko/followers
https://api.github.com/users/mariosasko/following{/other_user}
https://api.github.com/users/mariosasko/gists{/gist_id}
https://github.com/mariosasko
mariosasko
MDQ6VXNlcjQ3NDYyNzQy
https://api.github.com/users/mariosasko/orgs
https://api.github.com/users/mariosasko/received_events
https://api.github.com/users/mariosasko/repos
https://api.github.com/users/mariosasko/starred{/owner}{/repo}
https://api.github.com/users/mariosasko/subscriptions
User
https://api.github.com/users/mariosasko
public
NONE
I still think this is useful, since it's common that the data processing is different for training/dev/testing. And I don't know if the fact that `map` currently takes many arguments is a good reason not to support a useful feature.
2022-10-05T13:47:28Z
https://github.com/huggingface/datasets/issues/1593#issuecomment-1268463963
https://api.github.com/repos/huggingface/datasets/issues/1593
IC_kwDODunzps5Lmzlb
https://api.github.com/repos/huggingface/datasets/issues/comments/1268463963/reactions
2022-10-05T13:47:28Z
https://api.github.com/repos/huggingface/datasets/issues/comments/1268463963
https://avatars.githubusercontent.com/u/11954789?v=4
https://api.github.com/users/ZhaofengWu/events{/privacy}
https://api.github.com/users/ZhaofengWu/followers
https://api.github.com/users/ZhaofengWu/following{/other_user}
https://api.github.com/users/ZhaofengWu/gists{/gist_id}
https://github.com/ZhaofengWu
ZhaofengWu
MDQ6VXNlcjExOTU0Nzg5
https://api.github.com/users/ZhaofengWu/orgs
https://api.github.com/users/ZhaofengWu/received_events
https://api.github.com/users/ZhaofengWu/repos
https://api.github.com/users/ZhaofengWu/starred{/owner}{/repo}
https://api.github.com/users/ZhaofengWu/subscriptions
User
https://api.github.com/users/ZhaofengWu
public
It is possible that we want to do different things in the `map` function (and possibly other functions too) of a `DatasetDict`, depending on the key. I understand that `DatasetDict.map` is a really thin wrapper of `Dataset.map`, so it is easy to directly implement this functionality in the client code. Still, it'd be nice if there can be a flag, similar to `with_indices`, that allows the callable to know the key inside `DatasetDict`.