add requirements.txt

GoogleCloudPlatform · Jul 19, 2018 · b31572d · b31572d
1 parent 76c10c2
commit b31572d
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 60 deletions.
diff --git a/language/automl/automl_natural_language_model.py b/language/automl/automl_natural_language_model.py
@@ -54,28 +54,9 @@ def create_model(project_id, compute_region, dataset_id, model_name):
 
     # Create a model with the model metadata in the region.
     response = client.create_model(project_location, my_model)
-
     print("Training operation name: {}".format(response.operation.name))
     print("Training started...")
 
-    # synchronous check of operation status.
-    model = response.result()
-
-    # Retrieve deployment state.
-    if model.deployment_state == enums.Model.DeploymentState.DEPLOYED:
-        deployment_state = "deployed"
-    else:
-        deployment_state = "undeployed"
-
-    # Display the model information.
-    print("Model name: {}".format(model.name))
-    print("Model id: {}".format(model.name.split("/")[-1]))
-    print("Model display name: {}".format(model.display_name))
-    print("Model create time:")
-    print("\tseconds: {}".format(model.create_time.seconds))
-    print("\tnanos: {}".format(model.create_time.nanos))
-    print("Model deployment state: {}".format(deployment_state))
-
 
 # [END automl_natural_language_create_model]
 
@@ -91,7 +72,9 @@ def get_operation_status(operation_full_id):
     client = automl.AutoMlClient()
 
     # Get the latest state of a long-running operation.
-    response = client._operations_client.get_operation(operation_full_id)
+    response = client.transport._operations_client.get_operation(
+        operation_full_id
+    )
 
     print("Operation status: {}".format(response))
 

diff --git a/language/automl/requirements.txt b/language/automl/requirements.txt
@@ -0,0 +1 @@
+google-cloud-automl==0.1.0
diff --git a/translate/automl/automl_translation_model.py b/translate/automl/automl_translation_model.py
@@ -216,7 +216,9 @@ def get_operation_status(operation_full_id):
     client = automl.AutoMlClient()
 
     # Get the latest state of a long-running operation.
-    response = client._operations_client.get_operation(operation_full_id)
+    response = client.transport._operations_client.get_operation(
+        operation_full_id
+    )
 
     print("Operation status: {}".format(response))
 

diff --git a/translate/automl/requirements.txt b/translate/automl/requirements.txt
@@ -0,0 +1 @@
+google-cloud-automl==0.1.0
diff --git a/vision/automl/automl_vision_model.py b/vision/automl/automl_vision_model.py
@@ -63,44 +63,6 @@ def create_model(
     print("Training operation name: {}".format(response.operation.name))
     print("Training started...")
 
-    # synchronous check of operation status.
-    model = response.result()
-
-    # Retrieve deployment state.
-    if model.deployment_state == enums.Model.DeploymentState.DEPLOYED:
-        deployment_state = "deployed"
-    else:
-        deployment_state = "undeployed"
-    # Display the model information.
-    print("Model name: {}".format(model.name))
-    print("Model id: {}".format(model.name.split("/")[-1]))
-    print("Model display name: {}".format(model.display_name))
-    print("Image classification model metadata:")
-    print(
-        "Training budget: {}".format(
-            model.image_classification_model_metadata.train_budget
-        )
-    )
-    print(
-        "Training cost: {}".format(
-            model.image_classification_model_metadata.train_cost
-        )
-    )
-    print(
-        "Stop reason: {}".format(
-            model.image_classification_model_metadata.stop_reason
-        )
-    )
-    print(
-        "Base model id: {}".format(
-            model.image_classification_model_metadata.base_model_id
-        )
-    )
-    print("Model create time:")
-    print("\tseconds: {}".format(model.create_time.seconds))
-    print("\tnanos: {}".format(model.create_time.nanos))
-    print("Model deployment state: {}".format(deployment_state))
-
 
 # [END automl_vision_create_model]
 
@@ -116,7 +78,9 @@ def get_operation_status(operation_full_id):
     client = automl.AutoMlClient()
 
     # Get the latest state of a long-running operation.
-    response = client._operations_client.get_operation(operation_full_id)
+    response = client.transport._operations_client.get_operation(
+        operation_full_id
+    )
 
     print("Operation status: {}".format(response))
 

diff --git a/vision/automl/requirements.txt b/vision/automl/requirements.txt
@@ -0,0 +1 @@
+google-cloud-automl==0.1.0