nodes/HowToSD-ComfyUI-Pt-Wrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated readme.

Browse Source
This commit is contained in:
HowToSD
2025-03-28 21:23:24 -07:00
parent bda7a0b45b
commit c5a75142a2
2 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dev/scripts/generate_node_breakdown_csv.py
+4 -10
View File
@@ -25,19 +25,13 @@ def process_import_lines(file_path: str) -> Dict[str, int]:
with open(file_path, "r") as f:
for line in f:
if "import" in line:
if line.startswith("from typing"):
continue
module_name = re.sub(r"^.*import ", "", line).strip()
if module_name.startswith("SNS"):
increment_count(category_counts, "Seaborn")
elif module_name.startswith("MPL"):
increment_count(category_counts, "Matplotlib")
if module_name.startswith("SP"):
increment_count(category_counts, "SentencePiece")
elif module_name.startswith("Pt"):
increment_count(category_counts, "PyTorch")
elif module_name.startswith("Pandas"):
increment_count(category_counts, "Pandas")
elif module_name.startswith(("CDA", "Python")):
increment_count(category_counts, "Misc")
elif module_name.startswith("Numpy"):
increment_count(category_counts, "NumPy")
else:
increment_count(category_counts, "Misc")