Implementing unit tests for pytorch models

Hi Ravin,
Thanks for educating us in py test.
I have a NER model trained and saved in my local in .pkl format. I use that model for prediction (Its a pytorch model).
Now I want to implement the unit test cases for the prediction function using mock . How can I do it?