akseljoonas HF Staff commited on
Commit
6a4fd73
·
1 Parent(s): e82b057
Files changed (1) hide show
  1. agent/tools/dataset_tools.py +7 -7
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 = ["## Structure (configs & splits)", "| Config | Split |", "|--------|-------|"]
 
 
 
 
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