Skip to content

Commit

Permalink
docs: move and update beta list samples for video (#344)
Browse files Browse the repository at this point in the history
  • Loading branch information
nnegrey authored and Ace Nassri committed Nov 17, 2022
1 parent 4155e8a commit 25b17b2
Show file tree
Hide file tree
Showing 4 changed files with 209 additions and 0 deletions.
73 changes: 73 additions & 0 deletions automl/beta/list-datasets.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,73 @@
// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

'use strict';

function main(projectId = 'YOUR_PROJECT_ID', location = 'us-central1') {
// [START automl_video_classification_list_datasets_beta]
// [START automl_video_object_tracking_list_datasets_beta]
/**
* TODO(developer): Uncomment these variables before running the sample.
*/
// const projectId = 'YOUR_PROJECT_ID';
// const location = 'us-central1';

// Imports the Google Cloud AutoML library
const {AutoMlClient} = require(`@google-cloud/automl`).v1beta1;

// Instantiates a client
const client = new AutoMlClient();

async function listDatasets() {
// Construct request
const request = {
parent: client.locationPath(projectId, location),
filter: 'translation_dataset_metadata:*',
};

const [response] = await client.listDatasets(request);

console.log(`List of datasets:`);
for (const dataset of response) {
console.log(`Dataset name: ${dataset.name}`);
console.log(
`Dataset id: ${
dataset.name.split('/')[dataset.name.split('/').length - 1]
}`
);
console.log(`Dataset display name: ${dataset.displayName}`);
console.log(`Dataset create time`);
console.log(`\tseconds ${dataset.createTime.seconds}`);
console.log(`\tnanos ${dataset.createTime.nanos / 1e9}`);

// [END automl_video_object_tracking_list_datasets_beta]
console.log(
`Video classification dataset metadata: ${dataset.videoClassificationDatasetMetadata}`
);
// [END automl_video_classification_list_datasets_beta]

// [START automl_video_object_tracking_list_datasets_beta]
console.log(
`Video object tracking dataset metadata: ${dataset.videoObjectTrackingDatasetMetadata}`
);
// [START automl_video_classification_list_datasets_beta]
}
}

listDatasets();
// [END automl_video_classification_list_datasets_beta]
// [END automl_video_object_tracking_list_datasets_beta]
}

main(...process.argv.slice(2));
57 changes: 57 additions & 0 deletions automl/beta/list-models.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,57 @@
// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

'use strict';

function main(projectId = 'YOUR_PROJECT_ID', location = 'us-central1') {
// [START automl_list_models_beta]
/**
* TODO(developer): Uncomment these variables before running the sample.
*/
// const projectId = 'YOUR_PROJECT_ID';
// const location = 'us-central1';

// Imports the Google Cloud AutoML library
const {AutoMlClient} = require(`@google-cloud/automl`).v1beta1;

// Instantiates a client
const client = new AutoMlClient();

async function listModels() {
// Construct request
const request = {
parent: client.locationPath(projectId, location),
filter: 'translation_model_metadata:*',
};

const [response] = await client.listModels(request);

console.log(`List of models:`);
for (const model of response) {
console.log(`Model name: ${model.name}`);
console.log(`
Model id: ${model.name.split('/')[model.name.split('/').length - 1]}`);
console.log(`Model display name: ${model.displayName}`);
console.log(`Model create time`);
console.log(`\tseconds ${model.createTime.seconds}`);
console.log(`\tnanos ${model.createTime.nanos / 1e9}`);
console.log(`Model deployment state: ${model.deploymentState}`);
}
}

listModels();
// [END automl_list_models_beta]
}

main(...process.argv.slice(2));
39 changes: 39 additions & 0 deletions automl/test/list-datasets.beta.test.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

'use strict';

const {assert} = require('chai');
const {describe, it} = require('mocha');
const {AutoMlClient} = require('@google-cloud/automl').v1beta1;

const cp = require('child_process');

const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});

const LIST_DATASET_REGION_TAG = 'beta/list-datasets';
const LOCATION = 'us-central1';

describe('Automl List Dataset Tests', () => {
const client = new AutoMlClient();

it('should list datasets', async () => {
const projectId = await client.getProjectId();
const list_output = execSync(
`node ${LIST_DATASET_REGION_TAG}.js ${projectId} ${LOCATION}`
);

assert.match(list_output, /Dataset id/);
});
});
40 changes: 40 additions & 0 deletions automl/test/list-models.beta.test.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

'use strict';

const {assert} = require('chai');
const {describe, it} = require('mocha');
const {AutoMlClient} = require('@google-cloud/automl').v1beta1;

const cp = require('child_process');

const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});

const LIST_MODEL_REGION_TAG = 'beta/list-models';
const LOCATION = 'us-central1';

describe('Automl List Model Tests', () => {
const client = new AutoMlClient();

it('should list models', async () => {
const projectId = await client.getProjectId();

// list models
const list_model_output = execSync(
`node ${LIST_MODEL_REGION_TAG}.js ${projectId} ${LOCATION}`
);
assert.match(list_model_output, /Model id:/);
});
});

0 comments on commit 25b17b2

Please sign in to comment.