fix test

Author: b-yogesh

{{ cookiecutter.folder_name }}/tests/train/test_train.py

@@ -1,7 +1,7 @@
 # Hi, I am a test file. Please update me in the required places after you
 # have updated your package.
 
-from unittest.mock import MagicMock, patch
+from unittest.mock import MagicMock
 
 from {{cookiecutter.package_name}}.train.change_me_train import Trainer