diff --git a/reflex/utils/serializers.py b/reflex/utils/serializers.py index d3dbb1d4c..aa84f49e3 100644 --- a/reflex/utils/serializers.py +++ b/reflex/utils/serializers.py @@ -39,10 +39,24 @@ SERIALIZERS: dict[Type, Serializer] = {} SERIALIZER_TYPES: dict[Type, Type] = {} +@overload +def serializer( + fn: None = None, + to: Any = None, +) -> Serializer: ... + + +@overload +def serializer( + fn: Serializer, + to: None = None, +) -> functools.partial[Serializer]: ... + + def serializer( fn: Serializer | None = None, - to: Type | None = None, -) -> Serializer: + to: Any = None, +) -> Serializer | functools.partial[Serializer]: """Decorator to add a serializer for a given type. Args: