It would be neat to support Polars out of the box.

Probably don't need to change much under the hood. Just check if user input for data a Pandas. If it is not, try to import Polars and call pl.from_pandas() if it is available (raise an ImportError if not available).