test

Author: jtsodapop

.pre-commit-config.yaml

@@ -5,10 +5,11 @@ repos:
     hooks:
       - id: trailing-whitespace
       - id: end-of-file-fixer
+
   - repo: https://github.com/google/yapf
     rev: v0.31.0
     hooks:
-      id: yapf
-      name: "yapf (for labelbox-python repo)"
-      args: ['-i --verbose --recursive --parallel --style "google" ']
-      language_version: python3.8
+      - id: yapf
+        name: "yapf (for labelbox-python repo)"
+        args: ['-i --verbose --recursive --parallel --style "google" ']
+        language_version: python3.8

labelbox/client.py

@@ -30,13 +30,12 @@
 
 
 class Client:
-    """ A Labelbox client. 
+    """ A Labelbox client.
 
     Contains info necessary for connecting to a Labelbox server (URL,
     authentication key). Provides functions for querying and creating
     top-level data objects (Projects, Datasets).
     """
-
     def __init__(self,
                  api_key=None,
                  endpoint='https://api.labelbox.com/graphql',
@@ -133,7 +132,8 @@ def convert_value(value):
         if query is not None:
             if params is not None:
                 params = {
-                    key: convert_value(value) for key, value in params.items()
+                    key: convert_value(value)
+                    for key, value in params.items()
                 }
             data = json.dumps({
                 'query': query,
@@ -334,18 +334,18 @@ def upload_data(self,
 
         request_data = {
             "operations":
-                json.dumps({
-                    "variables": {
-                        "file": None,
-                        "contentLength": len(content),
-                        "sign": sign
-                    },
-                    "query":
-                        """mutation UploadFile($file: Upload!, $contentLength: Int!,
+            json.dumps({
+                "variables": {
+                    "file": None,
+                    "contentLength": len(content),
+                    "sign": sign
+                },
+                "query":
+                """mutation UploadFile($file: Upload!, $contentLength: Int!,
                                             $sign: Boolean) {
                             uploadFile(file: $file, contentLength: $contentLength,
                                        sign: $sign) {url filename} } """,
-                }),
+            }),
             "map": (None, json.dumps({"1": ["variables.file"]})),
         }
         response = requests.post(
@@ -354,7 +354,7 @@ def upload_data(self,
             data=request_data,
             files={
                 "1": (filename, content, content_type) if
-                     (filename and content_type) else content
+                (filename and content_type) else content
             })
 
         if response.status_code == 502:
@@ -527,7 +527,9 @@ def _create(self, db_object_type, data):
         res = res["create%s" % db_object_type.type_name()]
         return db_object_type(self, res)
 
-    def create_dataset(self, iam_integration=IAMIntegration._DEFAULT, **kwargs):
+    def create_dataset(self,
+                       iam_integration=IAMIntegration._DEFAULT,
+                       **kwargs):
         """ Creates a Dataset object on the server.
 
         Attribute values are passed as keyword arguments.
@@ -703,7 +705,7 @@ def get_data_row_ids_for_external_ids(
             for row in self.execute(
                     query_str,
                 {'externalId_in': external_ids[i:i + max_ids_per_request]
-                })['externalIdsToDataRowIds']:
+                 })['externalIdsToDataRowIds']:
                 result[row['externalId']].append(row['dataRowId'])
         return result