metatensor / featomic

Computing representations for atomistic machine learning
http://metatensor.github.io/featomic/
BSD 3-Clause "New" or "Revised" License
44 stars 13 forks source link

Allow a subset of dimensions to be passed to `selected_keys` argument in base calculators #321

Closed jwa7 closed 2 months ago

jwa7 commented 3 months ago

Currently, the user needs to know a priori the output keys returned by a given calculator's compute function to make a valid selection using the selected_keys argument. This is inconvenient for controlling, for instance, computation of blocks for a given subset of atom types.

It would be useful to be able to make key selections on a subset of dimensions. This is similar to the issue raised in #312 , but is requested for all calculators in general.