Fixed ML workflow and packaging

[chinmayshah99]

Checklist

• [ ] I have followed the Contribution Guidelines and Code of Conduct
• [ ] I have commented my code following the OpenMined Styleguide
• [ ] I have labeled this PR with the relevant Type labels
• [ ] My changes are covered by tests

[review-notebook-app[bot]]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB