From d4dd72469e03fb1a9ba9258af81e27c1dbc78ee4 Mon Sep 17 00:00:00 2001
From: Karma Riuk <riukkarma@gmail.com>
Date: Tue, 1 Apr 2025 14:46:42 +0200
Subject: [PATCH] instead of creating a list of the comments, using the
 paginated list and totalCount

---
 pull_requests.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pull_requests.py b/pull_requests.py
index 982c288..1bbca40 100644
--- a/pull_requests.py
+++ b/pull_requests.py
@@ -254,8 +254,8 @@ def process_pull(
     )
     dataset.entries.append(entry)
 
-    comments = list(pr.get_review_comments())
-    assert len(comments) == 1
+    comments = pr.get_review_comments()
+    assert comments.totalCount == 1
     comment = comments[0]
     commented_file_path = comment.path