feat: Add display tuning job button for Ipython environments when get…

…ting an existing job PiperOrigin-RevId: 647386006
googleapis · Jun 27, 2024 · 872b455 · 872b455
1 parent ef5aeda
commit 872b455
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/vertexai/tuning/_supervised_tuning.py b/vertexai/tuning/_supervised_tuning.py
@@ -102,4 +102,6 @@ def train(
 
 
 class SupervisedTuningJob(_tuning.TuningJob):
-    pass
+    def __init__(self, tuning_job_name: str):
+        super().__init__(tuning_job_name=tuning_job_name)
+        _ipython_utils.display_model_tuning_button(self)