From c5a75142a29b31dd93b356a0d32d7260c6d27d10 Mon Sep 17 00:00:00 2001 From: HowToSD Date: Fri, 28 Mar 2025 21:23:24 -0700 Subject: [PATCH] Updated readme. --- README.md | 2 +- dev/scripts/generate_node_breakdown_csv.py | 14 ++++---------- 2 files changed, 5 insertions(+), 11 deletions(-) diff --git a/README.md b/README.md index 5fbed41..9d25017 100644 --- a/README.md +++ b/README.md @@ -21,7 +21,7 @@ Originally a focused spin-off of [ComfyUI-Data-Analysis](https://github.com/HowT - Use pre-built model nodes for major architecture such as ResNet, LSTM, GRU, Transformer, or create your own model from various model nodes. - Perform tensor operations like `add`, `gather`, `scatter`, `where`, etc. - Support for processing text (e.g. tokenization, embedding) to feed to a model - +- Featuring **200 nodes** as of March 28, 2025. --- ## Example Workflows diff --git a/dev/scripts/generate_node_breakdown_csv.py b/dev/scripts/generate_node_breakdown_csv.py index 2e32c55..d9e606f 100644 --- a/dev/scripts/generate_node_breakdown_csv.py +++ b/dev/scripts/generate_node_breakdown_csv.py @@ -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")