Tidy up

app.py

@@ -6,11 +6,11 @@ import requests
 
 from defaults import OWNER, REPO
 
-from build_saved_issues import get_issues
-from update_stored_issues import update_issues
 import build_issue_dict
 import build_embeddings
 import shutil
+from fetch import get_issues
+from update_stored_issues import update_issues
 
 
 

build_embeddings.py

@@ -87,7 +87,7 @@ def embed_issues(
                 logger.info(f"Skipping issue {issue_id} as it has no body")
                 continue
 
-            if issue_type == "pull_request" and "pull_request" not in issue:
+            if issue_type == "pull" and "pull_request" not in issue:
                 logger.info(f"Skipping issue {issue_id} as it is not a pull request")
                 continue
 

build_saved_issues.py → fetch.py

@@ -1,3 +1,7 @@
+"""
+Script to fetch issues from the transformers repo and save them to a json file
+"""
+
 import json
 
 import argparse

find_similar_issues.py

@@ -98,6 +98,7 @@ def get_similar_issues(issue_no, query, top_k=5, token=TOKEN, owner=OWNER, repo=
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
+    parser.add_argument("")
     parser.add_argument("--issue_no", type=int, default=None)
     parser.add_argument("--query", type=str, default=None)
     parser.add_argument("--top_k", type=int, default=5)