test on different environments

[dhruv5995]

• [ ] 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.