Fix flake after rebase

Author: franchuterivera

test/test_api/test_api.py

@@ -1,8 +1,8 @@
 import os
 import pickle
 import sys
-import unittest
 import time
+import unittest
 from test.test_api.api_utils import print_debug_information
 
 import numpy as np