diff --git a/natsort/natsort.py b/natsort/natsort.py index 8e3a7b5..0430a44 100644 --- a/natsort/natsort.py +++ b/natsort/natsort.py @@ -9,11 +9,16 @@ import platform from functools import partial from operator import itemgetter +from typing import Callable, Iterable, TypeVar + +from _typeshed import SupportsLessThan import natsort.compat.locale from natsort import utils from natsort.ns_enum import NS_DUMB, ns +_T = TypeVar("_T") + def decoder(encoding): """ @@ -212,7 +217,7 @@ def natsort_keygen(key=None, alg=ns.DEFAULT): """ -def natsorted(seq, key=None, reverse=False, alg=ns.DEFAULT): +def natsorted(seq: Iterable[_T], key: Callable[[_T], SupportsLessThan]=None, reverse=False, alg=ns.DEFAULT): """ Sorts an iterable naturally.