adding missing import datetime

andrewferlitsch · andrewferlitsch · commit d02c0e8d056b · 2019-05-31T11:42:03.000-07:00
diff --git a/language/automl/automl_natural_language_entity_dataset.py b/language/automl/automl_natural_language_entity_dataset.py
@@ -33,6 +33,7 @@ def create_dataset(project_id, compute_region, dataset_name):
     # compute_region = '[COMPUTE_REGION]'
     # dataset_name = '[DATASET_NAME]'
 
+    from datetime import datetime
     from google.cloud import automl_v1beta1 as automl
 
     client = automl.AutoMlClient()
@@ -67,6 +68,7 @@ def list_datasets(project_id, compute_region, filter_):
     # compute_region = '[COMPUTE_REGION]'
     # filter_ = 'filter expression here'
 
+    from datetime import datetime
     from google.cloud import automl_v1beta1 as automl
 
     client = automl.AutoMlClient()
@@ -97,6 +99,7 @@ def get_dataset(project_id, compute_region, dataset_id):
     # compute_region = '[COMPUTE_REGION]'
     # dataset_id = '[DATASET_ID]'
 
+    from datetime import datetime
     from google.cloud import automl_v1beta1 as automl
 
     client = automl.AutoMlClient()
diff --git a/language/automl/automl_natural_language_entity_model.py b/language/automl/automl_natural_language_entity_model.py
@@ -84,6 +84,7 @@ def list_models(project_id, compute_region, filter_):
     # compute_region = '[COMPUTE_REGION]'
     # filter_ = '[DATASET_ID]'
 
+    from datetime import datetime
     from google.cloud import automl_v1beta1 as automl
     from google.cloud.automl_v1beta1 import enums
 
@@ -122,6 +123,7 @@ def get_model(project_id, compute_region, model_id):
     # compute_region = '[COMPUTE_REGION]'
     # model_id = '[MODEL_ID]'
 
+    from datetime import datetime
     from google.cloud import automl_v1beta1 as automl
     from google.cloud.automl_v1beta1 import enums
 
