Henry65
/

RepoSim4Py

Feature Extraction

code-understanding

text-embeddings-inference

Model card Files Files and versions

HenryStephen commited on Mar 10, 2024

Commit

7b66718

·

1 Parent(s): e8f04b1

Update pipeline progress bar

Files changed (1) hide show

RepoPipeline.py +1 -1

RepoPipeline.py CHANGED Viewed

@@ -290,7 +290,7 @@ class RepoPipeline(Pipeline):
         model_outputs = []
         # The number of repository.
         num_texts = sum(
-            len(x["codes"]) + len(x["docs"] + len(x["requirements"]) + len(x["readmes"])) for x in extracted_infos)
         with tqdm(total=num_texts) as progress_bar:
             # For each repository
             for repo_info in extracted_infos:

         model_outputs = []
         # The number of repository.
         num_texts = sum(
+            len(x["codes"]) + len(x["docs"]) + len(x["requirements"]) + len(x["readmes"]) for x in extracted_infos)
         with tqdm(total=num_texts) as progress_bar:
             # For each repository
             for repo_info in extracted_infos: