fix tests for base api

Author: ravinkohli

test/test_api/test_base_api.py

@@ -1,7 +1,7 @@
 import logging
 import re
 import unittest
-from unittest.mock import MagicMock, patch
+from unittest.mock import MagicMock
 
 import numpy as np