diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
index 67acb6bcd71..e73bfa2ae42 100644
--- a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
+++ b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
@@ -165,14 +165,16 @@ def create_experiment(
                     )
                 )
 
-    # fmt: off
     def tune(
         self,
         # TODO (andreyvelich): How to be consistent with other APIs (name) ?
         name: str,
-        model_provider_parameters: Optional["HuggingFaceModelParams"] = None,  # noqa: F821
-        dataset_provider_parameters: Optional[Union[
-            "HuggingFaceDatasetParams", "S3DatasetParams"]] = None,  # noqa: F821
+        model_provider_parameters: Optional[
+            "HuggingFaceModelParams"  # noqa: F821
+        ] = None,
+        dataset_provider_parameters: Optional[
+            Union["HuggingFaceDatasetParams", "S3DatasetParams"]  # noqa: F821
+        ] = None,
         trainer_parameters: Optional["HuggingFaceTrainerParams"] = None,  # noqa: F821
         storage_config: Optional[Dict[str, Optional[Union[str, List[str]]]]] = {
             "size": constants.PVC_DEFAULT_SIZE,
@@ -205,8 +207,6 @@ def tune(
         pip_index_url: str = "https://pypi.org/simple",
         metrics_collector_config: Dict[str, Any] = {"kind": "StdOut"},
     ):
-        # fmt: on
-
         """
         Create HyperParameter Tuning Katib Experiment using one of the following
         options: