[Benchmarking-Py] Minor cleanups

Author: DEKHTIARJonathan

tftrt/benchmarking-python/tf_hub/albert/infer.py

@@ -17,13 +17,13 @@
 
 import os
 import sys
+
 import numpy as np
 import tensorflow as tf
+
 # Allow import of top level python files
 import inspect
 
-from dataloading import get_dataset_cola
-
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
@@ -35,6 +35,8 @@
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner
 
+from dataloading import get_dataset_cola
+
 
 class CommandLineAPI(BaseCommandLineAPI):