Search in sources :

Example 11 with DatasetName

use of com.google.cloud.automl.v1beta1.DatasetName in project java-automl by googleapis.

the class DeleteDatasetTest method setUp.

@Before
public void setUp() throws IOException {
    // Create a fake dataset to be deleted
    // Create a random dataset name with a length of 32 characters (max allowed by AutoML)
    // To prevent name collisions when running tests in multiple java versions at once.
    // AutoML doesn't allow "-", but accepts "_"
    String datasetName = String.format("test_%s", UUID.randomUUID().toString().replace("-", "_").substring(0, 26));
    try (AutoMlClient client = AutoMlClient.create()) {
        LocationName projectLocation = LocationName.of(PROJECT_ID, "us-central1");
        TextExtractionDatasetMetadata metadata = TextExtractionDatasetMetadata.newBuilder().build();
        Dataset dataset = Dataset.newBuilder().setDisplayName(datasetName).setTextExtractionDatasetMetadata(metadata).build();
        Dataset createdDataset = client.createDataset(projectLocation, dataset);
        String[] names = createdDataset.getName().split("/");
        datasetId = names[names.length - 1];
    }
    bout = new ByteArrayOutputStream();
    out = new PrintStream(bout);
    originalPrintStream = System.out;
    System.setOut(out);
}
Also used : PrintStream(java.io.PrintStream) Dataset(com.google.cloud.automl.v1beta1.Dataset) TextExtractionDatasetMetadata(com.google.cloud.automl.v1beta1.TextExtractionDatasetMetadata) ByteArrayOutputStream(java.io.ByteArrayOutputStream) AutoMlClient(com.google.cloud.automl.v1beta1.AutoMlClient) LocationName(com.google.cloud.automl.v1beta1.LocationName) Before(org.junit.Before)

Example 12 with DatasetName

use of com.google.cloud.automl.v1beta1.DatasetName in project java-automl by googleapis.

the class DeleteDataset method deleteDataset.

// Delete a dataset
static void deleteDataset(String projectId, String datasetId) throws IOException, ExecutionException, InterruptedException {
    // the "close" method on the client to safely clean up any remaining background resources.
    try (AutoMlClient client = AutoMlClient.create()) {
        // Get the full path of the dataset.
        DatasetName datasetFullId = DatasetName.of(projectId, "us-central1", datasetId);
        Empty response = client.deleteDatasetAsync(datasetFullId).get();
        System.out.format("Dataset deleted. %s%n", response);
    }
}
Also used : Empty(com.google.protobuf.Empty) DatasetName(com.google.cloud.automl.v1beta1.DatasetName) AutoMlClient(com.google.cloud.automl.v1beta1.AutoMlClient)

Aggregations

Empty (com.google.protobuf.Empty)7 AutoMlClient (com.google.cloud.automl.v1beta1.AutoMlClient)6 AutoMlClient (com.google.cloud.automl.v1.AutoMlClient)5 DatasetName (com.google.cloud.automl.v1.DatasetName)5 DatasetName (com.google.cloud.automl.v1beta1.DatasetName)5 GcsSource (com.google.cloud.automl.v1.GcsSource)2 InputConfig (com.google.cloud.automl.v1.InputConfig)2 GcsSource (com.google.cloud.automl.v1beta1.GcsSource)2 InputConfig (com.google.cloud.automl.v1beta1.InputConfig)2 TimeoutException (java.util.concurrent.TimeoutException)2 After (org.junit.After)2 RetrySettings (com.google.api.gax.retrying.RetrySettings)1 Dataset (com.google.cloud.automl.v1.Dataset)1 GcsDestination (com.google.cloud.automl.v1.GcsDestination)1 OperationMetadata (com.google.cloud.automl.v1.OperationMetadata)1 OutputConfig (com.google.cloud.automl.v1.OutputConfig)1 ListTableSpecsPagedResponse (com.google.cloud.automl.v1beta1.AutoMlClient.ListTableSpecsPagedResponse)1 AutoMlSettings (com.google.cloud.automl.v1beta1.AutoMlSettings)1 BigQuerySource (com.google.cloud.automl.v1beta1.BigQuerySource)1 Dataset (com.google.cloud.automl.v1beta1.Dataset)1