Henry65
/

RepoSim4Py

Feature Extraction

code-understanding

text-embeddings-inference

Model card Files Files and versions

Henry65 commited on Jul 24, 2023

Commit

bf6e4e8

·

1 Parent(s): fe12786

Update RepoPipeline.py

Files changed (1) hide show

RepoPipeline.py +1 -1

RepoPipeline.py CHANGED Viewed

@@ -143,7 +143,7 @@ def extract_information(repos, headers=None):
                 # 4. Extracting requirements.
                 elif member.name.endswith("requirements.txt") and member.isfile():
                     try:
-                        lines = tar.extractfile(member).decode("utf-8").readlines()
                         # extract readme
                         requirements_set = extract_requirements(lines)
                         repo_info["requirements"].update(requirements_set)

                 # 4. Extracting requirements.
                 elif member.name.endswith("requirements.txt") and member.isfile():
                     try:
+                        lines = tar.extractfile(member).readlines()
                         # extract readme
                         requirements_set = extract_requirements(lines)
                         repo_info["requirements"].update(requirements_set)