Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Yoan Di Cosmo commited on
Commit ·
e14c4ab
1
Parent(s): ce69c24
small modif
Browse files
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
|