feat: Add display tuning job button for Ipython environments when getting an existing job

vertex-sdk-bot · copybara-github · commit 872b455bcdda · 2024-06-27T11:29:21.000-07:00
PiperOrigin-RevId: 647386006
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)
