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")