fix aidge_learning version to 0.2.1

Only two fixes were applied to aidge_learning, not enough to update the version to 0.3.0