Skip to content

Commit d73ed84

Browse files
fixed according to commits
1 parent d69b357 commit d73ed84

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

aiplatform/src/test/java/aiplatform/CreateTrainingPipelineTextClassificationSampleTest.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
import org.junit.BeforeClass;
3333
import org.junit.Rule;
3434
import org.junit.Test;
35+
import org.junit.jupiter.api.Disabled;
3536
import org.junit.runner.RunWith;
3637
import org.junit.runners.JUnit4;
3738

@@ -88,6 +89,7 @@ public void tearDown()
8889
System.setOut(originalPrintStream);
8990
}
9091

92+
@Disabled // Sample should be migrated to Vertex AI https://softserve-jirasw.atlassian.net/browse/DSAM-173
9193
@Test
9294
public void testCreateTrainingPipelineTextClassificationSample() throws IOException {
9395
// Act

aiplatform/src/test/java/aiplatform/EmbeddingBatchSampleTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ public class EmbeddingBatchSampleTest extends TestCase {
3939
private static String BUCKET_NAME;
4040
private static final String GCS_SOURCE_URI =
4141
"gs://cloud-samples-data/generative-ai/embeddings/embeddings_input.jsonl";
42-
private static final String GCS_DESTINATION_OUTPUT_PREFIX =
42+
private static final String GCS_OUTPUT_URI =
4343
String.format("gs://%s/embedding_batch_output", BUCKET_NAME);
4444
private static final String MODEL_ID = "textembedding-gecko@003";
4545
static Storage storage;
@@ -75,7 +75,7 @@ public void testEmbeddingBatchSample() throws IOException {
7575

7676
BatchPredictionJob batchPredictionJob =
7777
EmbeddingBatchSample.embeddingBatchSample(PROJECT_ID, LOCATION, GCS_SOURCE_URI,
78-
GCS_DESTINATION_OUTPUT_PREFIX, MODEL_ID);
78+
GCS_OUTPUT_URI, MODEL_ID);
7979

8080
Assertions.assertNotNull(batchPredictionJob);
8181
assertTrue(batchPredictionJob.getDisplayName().contains("my embedding batch job "));

0 commit comments

Comments
 (0)