fix line limit

Author: sirtorry

language/automl/automl_natural_language_dataset.py

@@ -122,7 +122,9 @@ def get_dataset(project_id, compute_region, dataset_id):
     client = automl.AutoMlClient()
 
     # Get the full path of the dataset
-    dataset_full_id = client.dataset_path(project_id, compute_region, dataset_id)
+    dataset_full_id = client.dataset_path(
+        project_id, compute_region, dataset_id
+    )
 
     # Get complete detail of the dataset.
     dataset = client.get_dataset(dataset_full_id)
@@ -156,7 +158,9 @@ def import_data(project_id, compute_region, dataset_id, path):
     client = automl.AutoMlClient()
 
     # Get the full path of the dataset.
-    dataset_full_id = client.dataset_path(project_id, compute_region, dataset_id)
+    dataset_full_id = client.dataset_path(
+        project_id, compute_region, dataset_id
+    )
 
     # Get the multiple Google Cloud Storage URIs.
     input_uris = path.split(",")
@@ -185,7 +189,9 @@ def export_data(project_id, compute_region, dataset_id, output_uri):
     client = automl.AutoMlClient()
 
     # Get the full path of the dataset.
-    dataset_full_id = client.dataset_path(project_id, compute_region, dataset_id)
+    dataset_full_id = client.dataset_path(
+        project_id, compute_region, dataset_id
+    )
 
     # Set the output URI
     output_config = {"gcs_destination": {"output_uri_prefix": output_uri}}
@@ -212,7 +218,9 @@ def delete_dataset(project_id, compute_region, dataset_id):
     client = automl.AutoMlClient()
 
     # Get the full path of the dataset.
-    dataset_full_id = client.dataset_path(project_id, compute_region, dataset_id)
+    dataset_full_id = client.dataset_path(
+        project_id, compute_region, dataset_id
+    )
 
     # Delete a dataset.
     response = client.delete_dataset(dataset_full_id)
@@ -226,7 +234,8 @@ def delete_dataset(project_id, compute_region, dataset_id):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
-        description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter
+        description=__doc__,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
     )
     subparsers = parser.add_subparsers(dest="command")
 
@@ -245,14 +254,20 @@ def delete_dataset(project_id, compute_region, dataset_id):
         "filter_", nargs="?", default="text_classification_dataset_metadata:*"
     )
 
-    get_dataset_parser = subparsers.add_parser("get_dataset", help=get_dataset.__doc__)
+    get_dataset_parser = subparsers.add_parser(
+        "get_dataset", help=get_dataset.__doc__
+    )
     get_dataset_parser.add_argument("dataset_id")
 
-    import_data_parser = subparsers.add_parser("import_data", help=import_data.__doc__)
+    import_data_parser = subparsers.add_parser(
+        "import_data", help=import_data.__doc__
+    )
     import_data_parser.add_argument("dataset_id")
     import_data_parser.add_argument("path")
 
-    export_data_parser = subparsers.add_parser("export_data", help=export_data.__doc__)
+    export_data_parser = subparsers.add_parser(
+        "export_data", help=export_data.__doc__
+    )
     export_data_parser.add_argument("dataset_id")
     export_data_parser.add_argument("output_uri")
 
@@ -268,14 +283,18 @@ def delete_dataset(project_id, compute_region, dataset_id):
 
     if args.command == "create_dataset":
         multilabel = True if args.multilabel == "True" else False
-        create_dataset(project_id, compute_region, args.dataset_name, multilabel)
+        create_dataset(
+            project_id, compute_region, args.dataset_name, multilabel
+        )
     if args.command == "list_datasets":
         list_datasets(project_id, compute_region, args.filter_)
     if args.command == "get_dataset":
         get_dataset(project_id, compute_region, args.dataset_id)
     if args.command == "import_data":
         import_data(project_id, compute_region, args.dataset_id, args.path)
     if args.command == "export_data":
-        export_data(project_id, compute_region, args.dataset_id, args.output_uri)
+        export_data(
+            project_id, compute_region, args.dataset_id, args.output_uri
+        )
     if args.command == "delete_dataset":
         delete_dataset(project_id, compute_region, args.dataset_id)

language/automl/automl_natural_language_model.py

@@ -199,7 +199,9 @@ def list_model_evaluations(project_id, compute_region, model_id, filter_):
 
 
 # [START automl_natural_language_get_model_evaluation]
-def get_model_evaluation(project_id, compute_region, model_id, model_evaluation_id):
+def get_model_evaluation(
+    project_id, compute_region, model_id, model_evaluation_id
+):
     """Get model evaluation.
     Args:
         project_id: Id of the project.
@@ -322,7 +324,8 @@ def delete_model(project_id, compute_region, model_id):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
-        description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter
+        description=__doc__,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
     )
     subparsers = parser.add_subparsers(dest="command")
 
@@ -337,7 +340,9 @@ def delete_model(project_id, compute_region, model_id):
     )
     get_operation_status_parser.add_argument("operation_full_id")
 
-    list_models_parser = subparsers.add_parser("list_models", help=list_models.__doc__)
+    list_models_parser = subparsers.add_parser(
+        "list_models", help=list_models.__doc__
+    )
     list_models_parser.add_argument("filter_")
 
     get_model_parser = subparsers.add_parser(
@@ -349,7 +354,9 @@ def delete_model(project_id, compute_region, model_id):
         "list_model_evaluations", help=list_model_evaluations.__doc__
     )
     list_model_evaluations_parser.add_argument("model_id")
-    list_model_evaluations_parser.add_argument("filter_", nargs="?", default="")
+    list_model_evaluations_parser.add_argument(
+        "filter_", nargs="?", default=""
+    )
 
     get_model_evaluation_parser = subparsers.add_parser(
         "get_model_evaluation", help=get_model_evaluation.__doc__
@@ -374,20 +381,26 @@ def delete_model(project_id, compute_region, model_id):
     args = parser.parse_args()
 
     if args.command == "create_model":
-        create_model(project_id, compute_region, args.dataset_id, args.model_name)
+        create_model(
+            project_id, compute_region, args.dataset_id, args.model_name
+        )
     if args.command == "get_operation_status":
         get_operation_status(args.operation_full_id)
     if args.command == "list_models":
         list_models(project_id, compute_region, args.filter_)
     if args.command == "get_model":
         get_model(project_id, compute_region, args.model_id)
     if args.command == "list_model_evaluations":
-        list_model_evaluations(project_id, compute_region, args.model_id, args.filter_)
+        list_model_evaluations(
+            project_id, compute_region, args.model_id, args.filter_
+        )
     if args.command == "get_model_evaluation":
         get_model_evaluation(
             project_id, compute_region, args.model_id, args.model_evaluation_id
         )
     if args.command == "display_evaluation":
-        display_evaluation(project_id, compute_region, args.model_id, args.filter_)
+        display_evaluation(
+            project_id, compute_region, args.model_id, args.filter_
+        )
     if args.command == "delete_model":
         delete_model(project_id, compute_region, args.model_id)

language/automl/automl_natural_language_predict.py

@@ -45,7 +45,9 @@ def predict(project_id, compute_region, model_id, file_path):
     prediction_client = automl.PredictionServiceClient()
 
     # Get the full path of the model.
-    model_full_id = automl_client.model_path(project_id, compute_region, model_id)
+    model_full_id = automl_client.model_path(
+        project_id, compute_region, model_id
+    )
 
     # Read the file content for prediction.
     with open(file_path, "rb") as content_file:
@@ -69,7 +71,8 @@ def predict(project_id, compute_region, model_id, file_path):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
-        description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter
+        description=__doc__,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
     )
     subparsers = parser.add_subparsers(dest="command")
 

language/automl/dataset_test.py

@@ -53,13 +53,17 @@ def test_dataset_create_import_delete(capsys):
 
 def test_dataset_list_get(capsys):
     # list datasets
-    automl_natural_language_dataset.list_datasets(project_id, compute_region, "")
+    automl_natural_language_dataset.list_datasets(
+        project_id, compute_region, ""
+    )
     out, _ = capsys.readouterr()
     list_dataset_output = out.splitlines()
     assert "Dataset id: " in list_dataset_output[2]
 
     # get dataset
     dataset_id = list_dataset_output[2].split()[2]
-    automl_natural_language_dataset.get_dataset(project_id, compute_region, dataset_id)
+    automl_natural_language_dataset.get_dataset(
+        project_id, compute_region, dataset_id
+    )
     out, _ = capsys.readouterr()
     assert "Dataset name: " in out

language/automl/model_test.py

@@ -55,7 +55,9 @@ def test_model_list_get_evaluate(capsys):
 
     # get model
     model_id = list_models_output[2].split()[2]
-    automl_natural_language_model.get_model(project_id, compute_region, model_id)
+    automl_natural_language_model.get_model(
+        project_id, compute_region, model_id
+    )
     out, _ = capsys.readouterr()
     assert "Model name: " in out
 

translate/automl/automl_translation_dataset.py

@@ -46,7 +46,10 @@ def create_dataset(project_id, compute_region, dataset_name, source, target):
     project_location = client.location_path(project_id, compute_region)
 
     # Specify the source and target language.
-    dataset_metadata = {"source_language_code": source, "target_language_code": target}
+    dataset_metadata = {
+        "source_language_code": source,
+        "target_language_code": target,
+    }
     # Set dataset name and dataset metadata
     my_dataset = {
         "display_name": dataset_name,
@@ -131,7 +134,9 @@ def get_dataset(project_id, compute_region, dataset_id):
     client = automl.AutoMlClient()
 
     # Get the full path of the dataset
-    dataset_full_id = client.dataset_path(project_id, compute_region, dataset_id)
+    dataset_full_id = client.dataset_path(
+        project_id, compute_region, dataset_id
+    )
 
     # Get complete detail of the dataset.
     dataset = client.get_dataset(dataset_full_id)
@@ -172,7 +177,9 @@ def import_data(project_id, compute_region, dataset_id, path):
     client = automl.AutoMlClient()
 
     # Get the full path of the dataset.
-    dataset_full_id = client.dataset_path(project_id, compute_region, dataset_id)
+    dataset_full_id = client.dataset_path(
+        project_id, compute_region, dataset_id
+    )
 
     # Get the multiple Google Cloud Storage URIs
     input_uris = path.split(",")
@@ -200,7 +207,9 @@ def delete_dataset(project_id, compute_region, dataset_id):
     client = automl.AutoMlClient()
 
     # Get the full path of the dataset.
-    dataset_full_id = client.dataset_path(project_id, compute_region, dataset_id)
+    dataset_full_id = client.dataset_path(
+        project_id, compute_region, dataset_id
+    )
 
     # Delete a dataset.
     response = client.delete_dataset(dataset_full_id)
@@ -214,7 +223,8 @@ def delete_dataset(project_id, compute_region, dataset_id):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
-        description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter
+        description=__doc__,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
     )
     subparsers = parser.add_subparsers(dest="command")
 
@@ -230,7 +240,9 @@ def delete_dataset(project_id, compute_region, dataset_id):
     )
     list_datasets_parser.add_argument("filter", nargs="?", default="")
 
-    import_data_parser = subparsers.add_parser("import_data", help=import_data.__doc__)
+    import_data_parser = subparsers.add_parser(
+        "import_data", help=import_data.__doc__
+    )
     import_data_parser.add_argument("dataset_id")
     import_data_parser.add_argument("path")
 
@@ -239,7 +251,9 @@ def delete_dataset(project_id, compute_region, dataset_id):
     )
     delete_dataset_parser.add_argument("dataset_id")
 
-    get_dataset_parser = subparsers.add_parser("get_dataset", help=get_dataset.__doc__)
+    get_dataset_parser = subparsers.add_parser(
+        "get_dataset", help=get_dataset.__doc__
+    )
     get_dataset_parser.add_argument("dataset_id")
 
     project_id = os.environ["PROJECT_ID"]
@@ -249,7 +263,11 @@ def delete_dataset(project_id, compute_region, dataset_id):
 
     if args.command == "create_dataset":
         create_dataset(
-            project_id, compute_region, args.dataset_name, args.source, args.target
+            project_id,
+            compute_region,
+            args.dataset_name,
+            args.source,
+            args.target,
         )
     if args.command == "list_datasets":
         list_datasets(project_id, compute_region, args.filter)

translate/automl/automl_translation_model.py

@@ -156,7 +156,9 @@ def list_model_evaluations(project_id, compute_region, model_id, filter_):
 
 
 # [START automl_translation_get_model_evaluation]
-def get_model_evaluation(project_id, compute_region, model_id, model_evaluation_id):
+def get_model_evaluation(
+    project_id, compute_region, model_id, model_evaluation_id
+):
     """Get model evaluation.
     Args:
         project_id: Id of the project.
@@ -224,7 +226,8 @@ def get_operation_status(operation_full_id):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
-        description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter
+        description=__doc__,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
     )
     subparsers = parser.add_subparsers(dest="command")
 
@@ -246,15 +249,19 @@ def get_operation_status(operation_full_id):
     get_model_evaluation_parser.add_argument("model_id")
     get_model_evaluation_parser.add_argument("model_evaluation_id")
 
-    get_model_parser = subparsers.add_parser("get_model", help=get_model.__doc__)
+    get_model_parser = subparsers.add_parser(
+        "get_model", help=get_model.__doc__
+    )
     get_model_parser.add_argument("model_id")
 
     get_operation_status_parser = subparsers.add_parser(
         "get_operation_status", help=get_operation_status.__doc__
     )
     get_operation_status_parser.add_argument("operation_full_id")
 
-    list_models_parser = subparsers.add_parser("list_models", help=list_models.__doc__)
+    list_models_parser = subparsers.add_parser(
+        "list_models", help=list_models.__doc__
+    )
     list_models_parser.add_argument("filter", nargs="?", default="")
 
     delete_model_parser = subparsers.add_parser(
@@ -268,13 +275,17 @@ def get_operation_status(operation_full_id):
     args = parser.parse_args()
 
     if args.command == "create_model":
-        create_model(project_id, compute_region, args.dataset_id, args.model_name)
+        create_model(
+            project_id, compute_region, args.dataset_id, args.model_name
+        )
     if args.command == "list_models":
         list_models(project_id, compute_region, args.filter)
     if args.command == "get_model":
         get_model(project_id, compute_region, args.model_id)
     if args.command == "list_model_evaluations":
-        list_model_evaluations(project_id, compute_region, args.model_id, args.filter)
+        list_model_evaluations(
+            project_id, compute_region, args.model_id, args.filter
+        )
     if args.command == "get_model_evaluation":
         get_model_evaluation(
             project_id, compute_region, args.model_id, args.model_evaluation_id