chore: add report_metrics in post_gen.py.

Signed-off-by: Electronic-Waste <[email protected]>
kubeflow · Jul 12, 2024 · e73826f · e73826f
1 parent a0c8749
commit e73826f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/hack/gen-python-sdk/post_gen.py b/hack/gen-python-sdk/post_gen.py
@@ -41,6 +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 helper functions.\n")
         lines.append("import kubeflow.katib.api.search as search\n")
         lines.append("# Import Katib helper constants.\n")