diff --git a/conservancy_beancount/tools/sort_entries.py b/conservancy_beancount/tools/sort_entries.py index e50d3fd0f1cd5cd3463ae3058233e5803e37e296..9d22334bd002667c59d356667c9e5760172e9e00 100644 --- a/conservancy_beancount/tools/sort_entries.py +++ b/conservancy_beancount/tools/sort_entries.py @@ -22,7 +22,6 @@ from beancount import loader as bc_loader from beancount.parser import printer as bc_printer from typing import ( - Hashable, Optional, Sequence, TextIO, @@ -31,6 +30,7 @@ from ..beancount_types import ( Directive, Entries, Errors, + Sortable, ) from .. import cliutil @@ -56,7 +56,7 @@ def parse_arguments(arglist: Optional[Sequence[str]]=None) -> argparse.Namespace """) return parser.parse_args(arglist) -def entry_sorter(entry: Directive) -> Hashable: +def entry_sorter(entry: Directive) -> Sortable: type_name = type(entry).__name__ if type_name == 'Transaction': return (entry.date, type_name, entry.narration, entry.payee or '') # type:ignore[attr-defined]