save

Author: Adrian Chang

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

@@ -306,8 +306,8 @@ def get_user_groups(self) -> Iterator["UserGroup"]:
             Iterator[UserGroup]: An iterator over the user groups.
         """
         query = """
-            query GetUserGroupsPyApi {
-                userGroups {
+            query GetUserGroupsPyApi($first: PageSize, $after: String) {
+                userGroups(first: $first, after: $after) {
                     nodes {
                         id
                         name

libs/labelbox/tests/conftest.py

@@ -91,7 +91,9 @@ def ephemeral_endpoint() -> str:
 
 
 def graphql_url(environ: str) -> str:
-    if environ == Environ.PROD:
+    if environ == Environ.LOCAL:
+        return 'http://localhost:3000/graphql'
+    elif environ == Environ.PROD:
         return 'https://api.labelbox.com/graphql'
     elif environ == Environ.STAGING:
         return 'https://api.lb-stage.xyz/graphql'
@@ -107,7 +109,9 @@ def graphql_url(environ: str) -> str:
 
 
 def rest_url(environ: str) -> str:
-    if environ == Environ.PROD:
+    if environ == Environ.LOCAL:
+        return 'http://localhost:3000/api/v1'
+    elif environ == Environ.PROD:
         return 'https://api.labelbox.com/api/v1'
     elif environ == Environ.STAGING:
         return 'https://api.lb-stage.xyz/api/v1'