facebookresearch / mbrl-lib

Library for Model Based RL
MIT License
952 stars 154 forks source link

elites in iCEM optimizer are never reset #186

Open nuria95 opened 2 months ago

nuria95 commented 2 months ago

Steps to reproduce

When iCEM optimizer is initialised , self.elites is initialised to None. But after finishing the iCEM optimisation, the attribute is never reset back to None. Even more, this variable is never reset again (even after resetting the TrajectoryOptimizerAgent.

Expected Results

I would expect the self.elites parameter to be reset to None before the return in iCEM optimize method.