Yoan Di Cosmo commited on
Commit
e14c4ab
·
1 Parent(s): ce69c24

small modif

Browse files
Files changed (1) hide show
  1. agent/tools/dataset_tools.py +1 -5
agent/tools/dataset_tools.py CHANGED
@@ -350,11 +350,7 @@ def _format_messages_structure(messages_data: Any) -> str | None:
350
 
351
 
352
  def _format_parquet_files(data: dict, max_rows: int = 10) -> str | None:
353
- """Format parquet file info, return None if no files.
354
-
355
- We cap the number of rendered lines to keep output manageable for
356
- datasets with many parquet groups.
357
- """
358
  files = data.get("parquet_files", [])
359
  if not files:
360
  return None
 
350
 
351
 
352
  def _format_parquet_files(data: dict, max_rows: int = 10) -> str | None:
353
+ """Format parquet file info, return None if no files."""
 
 
 
 
354
  files = data.get("parquet_files", [])
355
  if not files:
356
  return None