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: ['--parallel --style "google" ']

labelbox/client.py

@@ -36,7 +36,6 @@ 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',
@@ -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.
@@ -547,11 +549,12 @@ def create_dataset(self, iam_integration=IAMIntegration._DEFAULT, **kwargs):
         """
         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
+        if iam_integration is None:
+            return dataset
 
         try:
             if not isinstance(iam_integration, IAMIntegration):
@@ -702,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