Skip to content

Commit e09858b

Browse files
committed
Fix tuned model tests
Change-Id: I5ace9222954be7d903ebbdabab9efc663fa79174
1 parent f1f4124 commit e09858b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

google/generativeai/types/model_types.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def idecode_time(parent: dict["str", Any], name: str):
143143

144144
def decode_tuned_model(tuned_model: protos.TunedModel | dict["str", Any]) -> TunedModel:
145145
if isinstance(tuned_model, protos.TunedModel):
146-
tuned_model = type(tuned_model).to_dict(tuned_model) # pytype: disable=attribute-error
146+
tuned_model = type(tuned_model).to_dict(tuned_model, including_default_value_fields=False) # pytype: disable=attribute-error
147147
tuned_model["state"] = to_tuned_model_state(tuned_model.pop("state", None))
148148

149149
base_model = tuned_model.pop("base_model", None)

0 commit comments

Comments
 (0)