| 57 | |
| 58 | |
| 59 | class Dataset(Generic[T_INPUT, T_TARGET], List[DataPiece[T_INPUT, T_TARGET]]): |
| 60 | def get_inputs(self) -> List[T_INPUT]: |
| 61 | return [item.input for item in self] |
| 62 | |
| 63 | def get_targets(self) -> List[T_TARGET]: |
| 64 | return [item.target for item in self] |
| 65 | |
| 66 | |
| 67 | class Task(Generic[T_INPUT, T_OUTPUT, T_TARGET]): |