Add other scores besides energy_distance to the multitreatment case

[EgorKraevTransferwise]

Describe the bug
A clear and concise description of what the bug is.

To Reproduce
Steps to reproduce the behavior:

1. Prepare the data as '...'
2. Run '....'
3. See error

Expected behavior
A clear and concise description of what you expected to happen.

Screenshots
If applicable, add screenshots to help explain your problem.

Your environment

• Version of auto-causality, e.g branch/commit #/release/tag
• Source type and setup
• Target type and setup

Additional context
Add any other context about the problem here.