reivew comments

Author: manibox

labelbox/schema/organization.py

@@ -44,7 +44,7 @@ def __init__(self, *args, **kwargs):
     users = Relationship.ToMany("User", False)
     projects = Relationship.ToMany("Project", True)
     webhooks = Relationship.ToMany("Webhook", False)
-    resourceTags = Relationship.ToMany("ResourceTag", False)
+    resourceTags = Relationship.ToMany("resource_tags", False)
 
     def invite_user(
             self,
@@ -161,7 +161,10 @@ def get_resource_tags(self) -> List[ResourceTag]:
         query_str = """query GetOrganizationResourceTagsPyApi{organization{resourceTag{%s}}}""" % (
             query.results_query_part(ResourceTag))
 
-        return self.client.execute(query_str)['organization']['resourceTag']
+        return [
+            ResourceTag(self.client, tag) for tag in self.client.execute(
+                query_str)['organization']['resourceTag']
+        ]
 
     def get_iam_integrations(self) -> List["IAMIntegration"]:
         """

labelbox/schema/project.py

@@ -134,7 +134,7 @@ def update_project_resource_tags(
         project_id_param = "projectId"
         tag_ids_param = "resourceTagIds"
 
-        query_str = """mutation CreateProjectResourceTagsPyApi($%s:ID!,$%s:[String!]) {
+        query_str = """mutation UpdateProjectResourceTagsPyApi($%s:ID!,$%s:[String!]) {
             project(where:{id:$%s}){updateProjectResourceTags(input:{%s:$%s}){%s}}}""" % (
             project_id_param, tag_ids_param, project_id_param, tag_ids_param,
             tag_ids_param, query.results_query_part(ResourceTag))
@@ -144,7 +144,10 @@ def update_project_resource_tags(
             tag_ids_param: resource_tag_ids
         })
 
-        return res["project"]["updateProjectResourceTags"]
+        return [
+            ResourceTag(self.client, tag)
+            for tag in res["project"]["updateProjectResourceTags"]
+        ]
 
     def labels(self, datasets=None, order_by=None) -> PaginatedCollection:
         """ Custom relationship expansion method to support limited filtering.