fix: update post_gen.py.

Signed-off-by: Electronic-Waste <[email protected]>
kubeflow · Jul 30, 2024 · 73af86d · 73af86d
1 parent 85d034f
commit 73af86d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hack/gen-python-sdk/post_gen.py b/hack/gen-python-sdk/post_gen.py
@@ -41,8 +41,8 @@ def _rewrite_helper(input_file, output_file, rewrite_rules):
  if (output_file == "sdk/python/v1beta1/kubeflow/katib/__init__.py"):
  lines.append("# Import Katib API client.\n")
  lines.append("from kubeflow.katib.api.katib_client import KatibClient\n")
- lines.append("# Import Katib report metrics functions")
- lines.append("from kubeflow.katib.api.report_metrics import report_metrics")
+ lines.append("# Import Katib report metrics functions\n")
+ lines.append("from kubeflow.katib.api.report_metrics import report_metrics\n")
  lines.append("# Import Katib helper functions.\n")
  lines.append("import kubeflow.katib.api.search as search\n")
  lines.append("# Import Katib helper constants.\n")