thread safe

Author: Matt Sokoloff

labelbox/orm/db_object.py

@@ -259,23 +259,11 @@ def delete(self):
 def beta(fn):
 
     def wrapper(self, *args, **kwargs):
-        if not isinstance(self, DbObject):
-            raise TypeError(
-                "Cannot decorate functions that are not functions of `DbOjects` with `beta` decorator"
-            )
         if not self.client.enable_beta:
             raise Exception(
                 f"This function {fn.__name__} relies on a beta feature in the api. This means that the interface could change."
                 " Set `enable_beta=True` in the client to enable use of these functions."
             )
-        execute_fn = self.client.execute
-        try:
-            self.client.execute = partial(execute_fn, beta=True)
-            result = fn(self, *args, **kwargs)
-            if isinstance(result, PaginatedCollection):
-                result.beta = True
-            return result
-        finally:
-            self.client.execute = execute_fn
+        return fn(self, *args, **kwargs)
 
     return wrapper

labelbox/pagination.py

@@ -17,7 +17,8 @@ def __init__(self,
                  params,
                  dereferencing,
                  obj_class,
-                 cursor_path=None):
+                 cursor_path=None,
+                 beta=False):
         """ Creates a PaginatedCollection.
 
         Args:
@@ -39,7 +40,7 @@ def __init__(self,
         self.params = params
         self.dereferencing = dereferencing
         self.obj_class = obj_class
-        self.beta = False
+        self.beta = beta
 
         self._fetched_all = False
         self._data = []

labelbox/schema/organization.py

@@ -103,20 +103,16 @@ def invite_user(
             "projectRoleId": project_role.role.uid
         } for project_role in _project_roles]
 
-        res = self.client.execute(
-            query_str,
-            {
-                data_param: [{
-                    "inviterId": self.client.get_user().uid,
-                    "inviteeEmail": email,
-                    "organizationId": self.uid,
-                    "organizationRoleId": role.uid,
-                    "projects": projects
-                }]
-            },
-        )
-        # We prob want to return an invite
-        # Could support bulk ops in the future
+        res = self.client.execute(query_str, {
+            data_param: [{
+                "inviterId": self.client.get_user().uid,
+                "inviteeEmail": email,
+                "organizationId": self.uid,
+                "organizationRoleId": role.uid,
+                "projects": projects
+            }]
+        },
+                                  beta=True)
         invite_response = res['createInvites'][0]['invite']
         return Invite(self.client, invite_response)
 
@@ -131,10 +127,10 @@ def invite_limit(self) -> InviteLimit:
     
         """
         org_id_param = "organizationId"
-        res = self.client.execute(
-            """query InvitesLimitPyApi($%s: ID!) {
+        res = self.client.execute("""query InvitesLimitPyApi($%s: ID!) {
             invitesLimit(where: {id: $%s}) { used limit remaining }
-        }""" % (org_id_param, org_id_param), {org_id_param: self.uid})
+        }""" % (org_id_param, org_id_param), {org_id_param: self.uid},
+                                  beta=True)
         return InviteLimit(
             **{utils.snake_case(k): v for k, v in res['invitesLimit'].items()})