adaptive-intelligent-robotics / QDax

Accelerated Quality-Diversity
https://qdax.readthedocs.io/en/latest/
MIT License
258 stars 42 forks source link

Change indexing in UnstructuredRepertoire intra batch comp #185

Open LisaCoiffard opened 2 months ago

LisaCoiffard commented 2 months ago

Related issues: N/A

Quick fix to select the current fitness in intra batch comp in the add method of unstructured repertoire. The current indexing will fail if any of the descriptors in the batch have nan values (e.g. when re-initialising the repertoire in AURORA, after training the auto-encoder).

This PR introduces:

Checks

Future improvements

[List here potential observations made and/or improvements that could be made in the future. If relevant, open issues for those.]

Lookatator commented 2 weeks ago

Hi @LisaCoiffard, thank you very much for the PR! Could you fix the style issue in the CI, so that I can merge it? :D