Luthaf / rascaline

Computing representations for atomistic machine learning
https://luthaf.fr/rascaline/
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

Open jwa7 opened 1 month ago

jwa7 commented 1 month 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.