Re-visit MLflow integration

[david-thrower]

Challenges:

1. Making sure that if this is run in the same environment, it will not conflict the existing DB, e.g. overwriting instead of appending.
2. race condition Investigate race conditions with using a self-contained hyperameter tuner such as Optuna #74