refactor gitlab script

1 files changed, 3 insertions, 3 deletions

diff --git a/gitlab/downloader.py b/gitlab/downloader.py
index 2b73f1cf..89c37d46 100755
--- a/gitlab/downloader.py
+++ b/gitlab/downloader.py
@@ -1,4 +1,4 @@
-from requests import get
+from requests import get, Response
 from description_parser import parse_description
 from output import output_issue
 
@@ -6,7 +6,7 @@ project_id = 11167699
 per_page = 100
 url = f"https://gitlab.com/api/v4/projects/{project_id}/issues?per_page={per_page}"
 
-def pages_iterator(first):
+def pages_iterator(first : Response):
     current = first
     while current.links.get('next'):
         current.raise_for_status()
@@ -16,7 +16,7 @@ def pages_iterator(first):
     yield current
 
 def main():
-    for response in pages_iterator(get(url = url)):
+    for response in pages_iterator(get(url)):
         print(f"Current page: {response.headers['x-page']}")
 
         data = response.json()