Use decorator for exports instead of __all__

[zshipko]

This would improve the experience of exporting functions by not having to add the function by name to __all__ manually