Only support 1 element type in `elements ml-splits` command

When the command was implemented, it was made to support multiple types, in case ML scientist need it.

After some feedback from them, they never use it, so we can safely remove that feature.