- [ ] ensure the plugin works on different types of OS and for different python version - [ ] ensure tests work for different model types such as scikit-learn, tensor flow, PyTorch, sparkML etc.
