Spaces:
Running
Running
Commit
·
6a4fd73
1
Parent(s):
e82b057
ruff
Browse files
agent/tools/dataset_tools.py
CHANGED
|
@@ -169,11 +169,13 @@ def _extract_configs(splits_data: dict) -> list[SplitConfig]:
|
|
| 169 |
return list(configs.values())
|
| 170 |
|
| 171 |
|
| 172 |
-
def _format_structure(
|
| 173 |
-
configs: list[SplitConfig], max_rows: int = 10
|
| 174 |
-
) -> str:
|
| 175 |
"""Format configs and splits as a markdown table."""
|
| 176 |
-
lines = [
|
|
|
|
|
|
|
|
|
|
|
|
|
| 177 |
|
| 178 |
total_splits = sum(len(cfg["splits"]) for cfg in configs)
|
| 179 |
added_rows = 0
|
|
@@ -378,9 +380,7 @@ def _format_parquet_files(data: dict, max_rows: int = 10) -> str | None:
|
|
| 378 |
shown += 1
|
| 379 |
|
| 380 |
if total_groups > shown:
|
| 381 |
-
lines.append(
|
| 382 |
-
f"- ... (_showing {shown} of {total_groups} parquet groups_)"
|
| 383 |
-
)
|
| 384 |
return "\n".join(lines)
|
| 385 |
|
| 386 |
|
|
|
|
| 169 |
return list(configs.values())
|
| 170 |
|
| 171 |
|
| 172 |
+
def _format_structure(configs: list[SplitConfig], max_rows: int = 10) -> str:
|
|
|
|
|
|
|
| 173 |
"""Format configs and splits as a markdown table."""
|
| 174 |
+
lines = [
|
| 175 |
+
"## Structure (configs & splits)",
|
| 176 |
+
"| Config | Split |",
|
| 177 |
+
"|--------|-------|",
|
| 178 |
+
]
|
| 179 |
|
| 180 |
total_splits = sum(len(cfg["splits"]) for cfg in configs)
|
| 181 |
added_rows = 0
|
|
|
|
| 380 |
shown += 1
|
| 381 |
|
| 382 |
if total_groups > shown:
|
| 383 |
+
lines.append(f"- ... (_showing {shown} of {total_groups} parquet groups_)")
|
|
|
|
|
|
|
| 384 |
return "\n".join(lines)
|
| 385 |
|
| 386 |
|