fix pagination

README.md

@@ -46,7 +46,7 @@ The goal is to foster community contributions by making the review process as le
 
 ```py
 # requirements.txt
-gradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.0
+gradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.1
 ```
 
 2. Set `HF_TOKEN` as a Space secret.
@@ -105,7 +105,7 @@ Add the following line to it:
 
 ```bash
 # requirements.txt
-gradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.0
+gradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.1
 ```
 
 ### 2. Add a user token as `HF_TOKEN` secret

RELEASE.md

@@ -1,3 +1,9 @@
+## 0.2.1
+
+- Install `huggingface_hub` dependency from source (commit [`7c74445903fe86f694ce6e42c386b7bebee94008`](https://github.com/huggingface/huggingface_hub/commit/7c74445903fe86f694ce6e42c386b7bebee94008)) in order to fix pagination when listing discussions from a repo. See [Pull Request #1962)(https://github.com/huggingface/huggingface_hub/pull/1962). Let's not forget to switch back to installing from PyPi once huggingface_hub `0.21.x` is released.
+- Run recovery in the background to avoid interrupting the main thread.
+- In recovery, loop only through pull requests instead of listing all discussions.
+
 ## 0.2.0
 
 - Move `gradio_space_ci` configuration to README.md metadata. Simplify integration (single `enable_space_ci` call).

requirements.txt

@@ -1,3 +1,4 @@
 gradio
-huggingface_hub>=0.20.2
+# TODO: switch to huggingface_hub>=0.21.0 once released
+huggingface_hub@git+https://github.com/huggingface/huggingface_hub@7c74445903fe86f694ce6e42c386b7bebee94008
 gradio-space-ci@git+https://huggingface.co/spaces/Wauplin/gradio-space-ci

src/gradio_space_ci/__init__.py

@@ -38,4 +38,4 @@ else:
     from .webhook import enable_space_ci  # noqa: F401
 
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"

src/gradio_space_ci/webhook.py

@@ -197,19 +197,18 @@ background_pool = ThreadPoolExecutor(max_workers=1)
 
 def recover_after_restart(space_id: str) -> None:
     print("Looping through PRs to check if any needs to be synced.")
-    for discussion in get_repo_discussions(repo_id=space_id, repo_type="space"):
-        if discussion.is_pull_request:
-            if discussion.status == "open":
-                if not is_pr_synced(space_id=space_id, pr_num=discussion.num):
-                    # Found a PR that is not yet synced
-                    print(f"Recovery. Found an open PR that is not synced: {discussion.url}. Syncing it.")
-                    background_pool.submit(sync_ci_space, space_id=space_id, pr_num=discussion.num)
-            if discussion.status == "merged" or discussion.status == "closed":
-                ci_space_id = _get_ci_space_id(space_id=space_id, pr_num=discussion.num)
-                if repo_exists(repo_id=ci_space_id, repo_type="space"):
-                    # Found a PR for which the CI space has not been deleted
-                    print(f"Recovery. Found a closed PR with an active CI space: {discussion.url}. Deleting it.")
-                    background_pool.submit(delete_ci_space, space_id=space_id, pr_num=discussion.num)
+    for discussion in get_repo_discussions(repo_id=space_id, repo_type="space", discussion_type="pull_request"):
+        if discussion.status == "open":
+            if not is_pr_synced(space_id=space_id, pr_num=discussion.num):
+                # Found a PR that is not yet synced
+                print(f"Recovery. Found an open PR that is not synced: {discussion.url}. Syncing it.")
+                background_pool.submit(sync_ci_space, space_id=space_id, pr_num=discussion.num)
+        if discussion.status == "merged" or discussion.status == "closed":
+            ci_space_id = _get_ci_space_id(space_id=space_id, pr_num=discussion.num)
+            if repo_exists(repo_id=ci_space_id, repo_type="space"):
+                # Found a PR for which the CI space has not been deleted
+                print(f"Recovery. Found a closed PR with an active CI space: {discussion.url}. Deleting it.")
+                background_pool.submit(delete_ci_space, space_id=space_id, pr_num=discussion.num)
 
 
 ###