test

Author: jtsodapop

.pre-commit-config.yaml

@@ -11,4 +11,4 @@ repos:
     hooks:
       - id: yapf
         name: "yapf (for labelbox-python repo)"
-        args: [--verbose --recursive --parallel --style "google" ]
+        args: [--verbose --recursive --parallel --style="google" ]

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,
@@ -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,9 +527,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.
@@ -549,7 +547,8 @@ def create_dataset(self,
         """
         dataset = self._create(Entity.Dataset, kwargs)
 
-        if iam_integration == IAMIntegration._DEFAULT: iam_integration = self.get_organization().get_default_iam_integration()
+        if iam_integration == IAMIntegration._DEFAULT:iam_integration = self.get_organization(
+            ).get_default_iam_integration()
 
         if iam_integration is None:
             return dataset
@@ -703,7 +702,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