Skip to content

Commit 308ba12

Browse files
committed
move extraction to "dataset" tool
1 parent 27ba58c commit 308ba12

File tree

5 files changed

+197
-636
lines changed

5 files changed

+197
-636
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
"""Dataset utilities for Cairo Coder."""
22

3-
from .analysis import DatasetAnalyzer, analyze_dataset
3+
from .analysis import DatasetAnalyzer
44

5-
__all__ = ["DatasetAnalyzer", "analyze_dataset"]
5+
__all__ = ["DatasetAnalyzer"]

python/src/cairo_coder_tools/datasets/analysis.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ def collect_counts(
215215
for fw in r["frameworks"]:
216216
frameworks_counts[fw] += 1
217217

218-
return category_counts, intent_counts, ves_by_category, frameworks_counts
218+
return category_counts, intent_counts, leaves_by_category, frameworks_counts
219219

220220

221221
def plot_barh(ax, labels: list[str], counts: list[int], title: str, xlabel: str, color: str):

0 commit comments

Comments
 (0)