diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py
index bf94d45d076..fef18adfa0f 100644
--- a/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py
+++ b/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py
@@ -4,11 +4,17 @@
 
 import kubeflow.katib.katib_api_pb2 as katib_api_pb2
 import pytest
-from kubeflow.katib import (KatibClient, V1beta1AlgorithmSpec,
-                            V1beta1Experiment, V1beta1ExperimentSpec,
-                            V1beta1FeasibleSpace, V1beta1ObjectiveSpec,
-                            V1beta1ParameterSpec, V1beta1TrialParameterSpec,
-                            V1beta1TrialTemplate)
+from kubeflow.katib import (
+    KatibClient,
+    V1beta1AlgorithmSpec,
+    V1beta1Experiment,
+    V1beta1ExperimentSpec,
+    V1beta1FeasibleSpace,
+    V1beta1ObjectiveSpec,
+    V1beta1ParameterSpec,
+    V1beta1TrialParameterSpec,
+    V1beta1TrialTemplate,
+)
 from kubeflow.katib.constants import constants
 from kubernetes.client import V1ObjectMeta