From 2863dc0ba2337a0e997b95e2cb8669abd62635e3 Mon Sep 17 00:00:00 2001 From: Yu-Han Liu Date: Mon, 8 Feb 2021 14:56:27 -0800 Subject: [PATCH] feat: LRO metadata (#204) * chore: release 0.5.0 Release-As: 0.5.0 * feat: update export model examples to print operation.metadata.output_info --- .sample_configs/process_configs.yaml | 20 ++++++++++++++----- samples/snippets/export_model_sample.py | 1 + ...ort_model_tabular_classification_sample.py | 1 + ...odel_tabular_classification_sample_test.py | 2 +- ...t_model_video_action_recognition_sample.py | 1 + ...el_video_action_recognition_sample_test.py | 2 +- 6 files changed, 20 insertions(+), 7 deletions(-) diff --git a/.sample_configs/process_configs.yaml b/.sample_configs/process_configs.yaml index e89a42e4c8..a671cf8710 100644 --- a/.sample_configs/process_configs.yaml +++ b/.sample_configs/process_configs.yaml @@ -148,11 +148,21 @@ export_data_sample: resource_name: dataset export_evaluated_data_items_sample: resource_name: model_evaluation -export_model_sample: {} -export_model_tabular_classification_sample: {} -export_model_video_action_recognition_sample: {} -export_model_video_classification_sample: {} -export_model_video_object_tracking_sample: {} +export_model_sample: + lro_metadata: + - output_info +export_model_tabular_classification_sample: + lro_metadata: + - output_info +export_model_video_action_recognition_sample: + lro_metadata: + - output_info +export_model_video_classification_sample: + lro_metadata: + - output_info +export_model_video_object_tracking_sample: + lro_metadata: + - output_info get_annotation_spec_sample: {} get_batch_prediction_job_sample: skip: diff --git a/samples/snippets/export_model_sample.py b/samples/snippets/export_model_sample.py index 42d8369909..ab066e99ed 100644 --- a/samples/snippets/export_model_sample.py +++ b/samples/snippets/export_model_sample.py @@ -35,6 +35,7 @@ def export_model_sample( name = client.model_path(project=project, location=location, model=model_id) response = client.export_model(name=name, output_config=output_config) print("Long running operation:", response.operation.name) + print("output_info:", response.metadata.output_info) export_model_response = response.result(timeout=timeout) print("export_model_response:", export_model_response) diff --git a/samples/snippets/export_model_tabular_classification_sample.py b/samples/snippets/export_model_tabular_classification_sample.py index 5da02ef5b8..596adb8a64 100644 --- a/samples/snippets/export_model_tabular_classification_sample.py +++ b/samples/snippets/export_model_tabular_classification_sample.py @@ -37,6 +37,7 @@ def export_model_tabular_classification_sample( name = client.model_path(project=project, location=location, model=model_id) response = client.export_model(name=name, output_config=output_config) print("Long running operation:", response.operation.name) + print("output_info:", response.metadata.output_info) export_model_response = response.result(timeout=timeout) print("export_model_response:", export_model_response) diff --git a/samples/snippets/export_model_tabular_classification_sample_test.py b/samples/snippets/export_model_tabular_classification_sample_test.py index e34bc08410..f74de8648c 100644 --- a/samples/snippets/export_model_tabular_classification_sample_test.py +++ b/samples/snippets/export_model_tabular_classification_sample_test.py @@ -42,4 +42,4 @@ def test_ucaip_generated_export_model_tabular_classification_sample(capsys): gcs_destination_output_uri_prefix=f"{GCS_BUCKET}/{GCS_PREFIX}", ) out, _ = capsys.readouterr() - assert "export_model_response" in out + assert "output_info" in out diff --git a/samples/snippets/export_model_video_action_recognition_sample.py b/samples/snippets/export_model_video_action_recognition_sample.py index 930a882c0a..f2f7ff1b86 100644 --- a/samples/snippets/export_model_video_action_recognition_sample.py +++ b/samples/snippets/export_model_video_action_recognition_sample.py @@ -38,6 +38,7 @@ def export_model_video_action_recognition_sample( name = client.model_path(project=project, location=location, model=model_id) response = client.export_model(name=name, output_config=output_config) print("Long running operation:", response.operation.name) + print("output_info:", response.metadata.output_info) export_model_response = response.result(timeout=timeout) print("export_model_response:", export_model_response) diff --git a/samples/snippets/export_model_video_action_recognition_sample_test.py b/samples/snippets/export_model_video_action_recognition_sample_test.py index cee43a2911..c690c466e5 100644 --- a/samples/snippets/export_model_video_action_recognition_sample_test.py +++ b/samples/snippets/export_model_video_action_recognition_sample_test.py @@ -46,4 +46,4 @@ def test_export_model_video_action_recognition_sample(capsys): export_format=EXPORT_FORMAT, ) out, _ = capsys.readouterr() - assert "export_model_response" in out + assert "output_info" in out