Use correct method to create object

Author: Adrian Chang

libs/labelbox/src/labelbox/exceptions.py

@@ -32,17 +32,21 @@ class AuthorizationError(LabelboxError):
 class ResourceNotFoundError(LabelboxError):
     """Exception raised when a given resource is not found. """
 
-    def __init__(self, db_object_type, params):
-        """ Constructor.
+    def __init__(self, db_object_type=None, params=None, message=None):
+        """Constructor for the ResourceNotFoundException class.
 
         Args:
-            db_object_type (type): A labelbox.schema.DbObject subtype.
-            params (dict): Dict of params identifying the sought resource.
+            db_object_type (type): A subtype of labelbox.schema.DbObject.
+            params (dict): A dictionary of parameters identifying the sought resource.
+            message (str): An optional message to include in the exception.
         """
-        super().__init__("Resource '%s' not found for params: %r" %
-                         (db_object_type.type_name(), params))
-        self.db_object_type = db_object_type
-        self.params = params
+        if message is not None:
+            super().__init__(message)
+        else:
+            super().__init__("Resource '%s' not found for params: %r" %
+                                (db_object_type.type_name(), params))
+            self.db_object_type = db_object_type
+            self.params = params
 
 
 class ResourceConflict(LabelboxError):

libs/labelbox/src/labelbox/schema/user_group.py

@@ -139,9 +139,7 @@ def get(self) -> "UserGroup":
         }
         result = self.client.execute(query, params)
         if not result:
-            error = ResourceNotFoundError.__new__(ResourceNotFoundError)
-            error.message = "Failed to get user group as user group does not exist"
-            raise error
+            raise ResourceNotFoundError(message="Failed to get user group as user group does not exist")
         self.name = result["userGroup"]["name"]
         self.color = UserGroupColor(result["userGroup"]["color"])
         self.projects = self._get_projects_set(result["userGroup"]["projects"]["nodes"])
@@ -202,9 +200,7 @@ def update(self) -> "UserGroup":
         try:
             result = self.client.execute(query, params)
             if not result:
-                error = ResourceNotFoundError.__new__(ResourceNotFoundError)
-                error.message = "Failed to update user group as user group does not exist"
-                raise error
+                raise ResourceNotFoundError(message="Failed to update user group as user group does not exist")
         except MalformedQueryException as e:
             raise UnprocessableEntityError("Failed to update user group") from e
         return self
@@ -303,9 +299,7 @@ def delete(self) -> bool:
         params = {"id": self.id}
         result = self.client.execute(query, params)
         if not result:
-            error = ResourceNotFoundError.__new__(ResourceNotFoundError)
-            error.message = "Failed to delete user group as user group does not exist"
-            raise error
+            raise ResourceNotFoundError(message="Failed to delete user group as user group does not exist")
         return result["deleteUserGroup"]["success"]
 
     def get_user_groups(self) -> Iterator["UserGroup"]: