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: fix sz on windows

Browse Source
This commit is contained in:
George Hotz
2024-10-21 16:02:23 +08:00
parent 87a1e76745
commit 17e7d8f10e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sz.py
View File

@@ -15,9 +15,9 @@ def gen_stats(base_path="."):
for path, _, files in os.walk(os.path.join(base_path, "tinygrad")):
for name in files:
if not name.endswith(".py"): continue
if 'tinygrad/runtime/autogen' in path: continue
if 'tinygrad/runtime/autogen' in path.replace('\\', '/'): continue
filepath = os.path.join(path, name)
relfilepath = os.path.relpath(filepath, base_path)
relfilepath = os.path.relpath(filepath, base_path).replace('\\', '/')
with tokenize.open(filepath) as file_:
tokens = [t for t in tokenize.generate_tokens(file_.readline) if t.type in TOKEN_WHITELIST and not is_docstring(t)]
token_count, line_count = len(tokens), len(set([x for t in tokens for x in range(t.start[0], t.end[0]+1)]))