test

Author: jtsodapop

labelbox/client.py

@@ -32,10 +32,10 @@
 class Client:
     """ A Labelbox client.
 
-    Contains info necessary for connecting to a Labelbox server (URL, authentication key). Provides functions for querying and creating
+    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',
@@ -132,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 +335,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 +355,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 +528,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 +706,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