Add retry for ml tests as they are flapping.

[chenkins]

Objectives
Describe the objectives this issue should fulfill.

Todos
List what needs to be done for the issue to be considered done and the objectives to be reached.

Minimal testing requirements
Think about what behavior shall be covered by a test (qualitatively describe each test).

Additional context
Add any other context or screenshots about the feature request here.

https://pypi.org/project/pytest-retry/