Setup the CI/CD yaml to test the train

• The python file can be executed correctly
• MLflow can store the parameters and models