dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-02 02:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

style: format code with linter

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-05 16:54:46 -03:00
parent 276b30506c
commit de26f8e5e0
1 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

15
aider/repomap.py
View file

@ -24,9 +24,10 @@ from aider.dump import dump # noqa: F402,E402
Tag = namedtuple("Tag", "rel_fname fname line name kind".split()) Tag = namedtuple("Tag", "rel_fname fname line name kind".split())
def print_elapsed(message): def print_elapsed(message):
current_time = time.time() current_time = time.time()
if hasattr(print_elapsed, 'last_time'): if hasattr(print_elapsed, "last_time"):
elapsed = current_time - print_elapsed.last_time elapsed = current_time - print_elapsed.last_time
print(f"{message}: {elapsed:.2f} seconds") print(f"{message}: {elapsed:.2f} seconds")
else: else:
@ -256,7 +257,7 @@ class RepoMap:
fnames = tqdm(fnames) fnames = tqdm(fnames)
self.cache_missing = False self.cache_missing = False
print_elapsed('Starting tags processing') print_elapsed("Starting tags processing")
for fname in fnames: for fname in fnames:
if not Path(fname).is_file(): if not Path(fname).is_file():
@ -294,7 +295,7 @@ class RepoMap:
elif tag.kind == "ref": elif tag.kind == "ref":
references[tag.name].append(rel_fname) references[tag.name].append(rel_fname)
print_elapsed('Finished tags processing, starting graph creation') print_elapsed("Finished tags processing, starting graph creation")
## ##
# dump(defines) # dump(defines)
# dump(references) # dump(references)
@ -335,13 +336,13 @@ class RepoMap:
else: else:
pers_args = dict() pers_args = dict()
print_elapsed('Starting pagerank calculation') print_elapsed("Starting pagerank calculation")
try: try:
ranked = nx.pagerank(G, weight="weight", **pers_args) ranked = nx.pagerank(G, weight="weight", **pers_args)
except ZeroDivisionError: except ZeroDivisionError:
return [] return []
print_elapsed('Starting rank distribution') print_elapsed("Starting rank distribution")
# distribute the rank from each source node, across all of its out edges # distribute the rank from each source node, across all of its out edges
ranked_definitions = defaultdict(float) ranked_definitions = defaultdict(float)
@ -359,7 +360,7 @@ class RepoMap:
# dump(ranked_definitions) # dump(ranked_definitions)
print_elapsed('Starting ranked tags processing') print_elapsed("Starting ranked tags processing")
for (fname, ident), rank in ranked_definitions: for (fname, ident), rank in ranked_definitions:
# print(f"{rank:.03f} {fname} {ident}") # print(f"{rank:.03f} {fname} {ident}")
if fname in chat_rel_fnames: if fname in chat_rel_fnames:
@ -380,7 +381,7 @@ class RepoMap:
for fname in rel_other_fnames_without_tags: for fname in rel_other_fnames_without_tags:
ranked_tags.append((fname,)) ranked_tags.append((fname,))
print_elapsed('Finished processing') print_elapsed("Finished processing")
return ranked_tags return ranked_tags
def get_ranked_tags_map( def get_ranked_tags_map(