test

Author: jtsodapop

labelbox/client.py

@@ -36,6 +36,7 @@ class Client:
     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,8 +133,7 @@ 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,
@@ -335,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(
@@ -355,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:
@@ -528,9 +528,7 @@ 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.
@@ -706,7 +704,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