Skip to content

Commit 9d15f09

Browse files
speedstorm1copybara-github
authored andcommitted
chore: Update tuning tests and the Tuning validation dataset field conversion
PiperOrigin-RevId: 822341715
1 parent 290e905 commit 9d15f09

File tree

3 files changed

+9
-12
lines changed

3 files changed

+9
-12
lines changed

src/main/java/com/google/genai/Tunings.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -850,8 +850,8 @@ ObjectNode tuningValidationDatasetToVertex(JsonNode fromObject, ObjectNode paren
850850

851851
if (Common.getValueByPath(fromObject, new String[] {"vertexDatasetResource"}) != null) {
852852
Common.setValueByPath(
853-
parentObject,
854-
new String[] {"supervisedTuningSpec", "trainingDatasetUri"},
853+
toObject,
854+
new String[] {"validationDatasetUri"},
855855
Common.getValueByPath(fromObject, new String[] {"vertexDatasetResource"}));
856856
}
857857

src/main/java/com/google/genai/types/CreateTuningJobConfig.java

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,7 @@ public abstract class CreateTuningJobConfig extends JsonSerializable {
3535
@JsonProperty("httpOptions")
3636
public abstract Optional<HttpOptions> httpOptions();
3737

38-
/**
39-
* Cloud Storage path to file containing training dataset for tuning. The dataset must be
40-
* formatted as a JSONL file.
41-
*/
38+
/** Validation dataset for tuning. The dataset must be formatted as a JSONL file. */
4239
@JsonProperty("validationDataset")
4340
public abstract Optional<TuningValidationDataset> validationDataset();
4441

@@ -138,17 +135,17 @@ public Builder httpOptions(HttpOptions.Builder httpOptionsBuilder) {
138135
/**
139136
* Setter for validationDataset.
140137
*
141-
* <p>validationDataset: Cloud Storage path to file containing training dataset for tuning. The
142-
* dataset must be formatted as a JSONL file.
138+
* <p>validationDataset: Validation dataset for tuning. The dataset must be formatted as a JSONL
139+
* file.
143140
*/
144141
@JsonProperty("validationDataset")
145142
public abstract Builder validationDataset(TuningValidationDataset validationDataset);
146143

147144
/**
148145
* Setter for validationDataset builder.
149146
*
150-
* <p>validationDataset: Cloud Storage path to file containing training dataset for tuning. The
151-
* dataset must be formatted as a JSONL file.
147+
* <p>validationDataset: Validation dataset for tuning. The dataset must be formatted as a JSONL
148+
* file.
152149
*/
153150
public Builder validationDataset(TuningValidationDataset.Builder validationDatasetBuilder) {
154151
return validationDataset(validationDatasetBuilder.build());

src/main/java/com/google/genai/types/TuningValidationDataset.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ public abstract class TuningValidationDataset extends JsonSerializable {
3434
public abstract Optional<String> gcsUri();
3535

3636
/**
37-
* The resource name of the Vertex Multimodal Dataset that is used as training dataset. Example:
37+
* The resource name of the Vertex Multimodal Dataset that is used as validation dataset. Example:
3838
* 'projects/my-project-id-or-number/locations/my-location/datasets/my-dataset-id'.
3939
*/
4040
@JsonProperty("vertexDatasetResource")
@@ -70,7 +70,7 @@ private static Builder create() {
7070
* Setter for vertexDatasetResource.
7171
*
7272
* <p>vertexDatasetResource: The resource name of the Vertex Multimodal Dataset that is used as
73-
* training dataset. Example:
73+
* validation dataset. Example:
7474
* 'projects/my-project-id-or-number/locations/my-location/datasets/my-dataset-id'.
7575
*/
7676
@JsonProperty("vertexDatasetResource")

0 commit comments

Comments
 (0)