made postfix better

This commit is contained in:
Karma Riuk
2025-03-20 10:13:35 +01:00
parent c5abfb457e
commit 8081b4ff76

View File

@ -193,8 +193,10 @@ def process_repo(repo_name: str, stats_df: Optional[pd.DataFrame], dataset: Data
repo = g.get_repo(repo_name) repo = g.get_repo(repo_name)
good_prs = get_good_prs(repo, stats_df) good_prs = get_good_prs(repo, stats_df)
for pr in tqdm(good_prs, desc="Processing good prs", leave=False): with tqdm(good_prs, desc="Processing good prs", leave=False) as pbar:
process_pull(repo, pr, dataset, repos_dir) for pr in pbar:
pbar.set_postfix({"pr": pr.number})
process_pull(repo, pr, dataset, repos_dir)
def process_repos(csv_file: str, stats_csv: Optional[str], dataset: Dataset, repos_dir: str): def process_repos(csv_file: str, stats_csv: Optional[str], dataset: Dataset, repos_dir: str):
""" """