Skip to content

Commit 5058422

Browse files
speedstorm1copybara-github
authored andcommitted
chore: Skip unnecessary conversion of training dataset object passed into Tunings.tune()
PiperOrigin-RevId: 822609104
1 parent ef98eae commit 5058422

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

google/genai/tunings.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -248,12 +248,8 @@ def _CreateTuningJobParametersPrivate_to_vertex(
248248
setv(to_object, ['preTunedModel'], getv(from_object, ['pre_tuned_model']))
249249

250250
if getv(from_object, ['training_dataset']) is not None:
251-
setv(
252-
to_object,
253-
['supervisedTuningSpec', 'trainingDatasetUri'],
254-
_TuningDataset_to_vertex(
255-
getv(from_object, ['training_dataset']), to_object, root_object
256-
),
251+
_TuningDataset_to_vertex(
252+
getv(from_object, ['training_dataset']), to_object, root_object
257253
)
258254

259255
if getv(from_object, ['config']) is not None:

0 commit comments

Comments
 (0)