Fix isort

ptflops/aten_engine.py

@@ -7,10 +7,10 @@
 '''
 
 
-from functools import partial
 import sys
 import traceback
 from collections import defaultdict
+from functools import partial
 from typing import Optional, Tuple, Union
 
 import torch