(dataset: list[dict], max_samples: int | None)
| 94 | |
| 95 | |
| 96 | def _limit_dataset(dataset: list[dict], max_samples: int | None) -> list[dict]: |
| 97 | if max_samples is None or len(dataset) <= max_samples: |
| 98 | return dataset |
| 99 | random.shuffle(dataset) |
| 100 | return dataset[:max_samples] |
| 101 | |
| 102 | |
| 103 | def _apply_chat_template(tokenizer, messages: list[dict], enable_thinking: bool) -> str: |
no outgoing calls
no test coverage detected