github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: make the line counter correct

Browse Source
This commit is contained in:
George Hotz
2024-01-01 11:01:22 -08:00
parent 56f44bd10e
commit 45247385eb
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sz.py
View File

@@ -16,8 +16,7 @@ def gen_stats(base_path="."):
relfilepath = os.path.relpath(filepath, base_path)
with tokenize.open(filepath) as file_:
tokens = [t for t in tokenize.generate_tokens(file_.readline) if t.type in TOKEN_WHITELIST]
#token_count, line_count = len(tokens), len(set([x for t in tokens for x in range(t.start[0], t.end[0]+1)]))
token_count, line_count = len(tokens), len(set([t.start[0] for t in tokens]))
token_count, line_count = len(tokens), len(set([x for t in tokens for x in range(t.start[0], t.end[0]+1)]))
table.append([relfilepath, line_count, token_count/line_count])
return table