Add timezone arg to st.date_input?

My dataframe has a datetime column as datetime64[ns, UTC]. When doing st.date_input and trying to use that to filter my dataframe I get TypeError: Invalid comparison between dtype=datetime64[ns, UTC] and datetime

Would be nice to have a tz arg to st.date_input.

I can probably just use pd.to_datetime when I filter with it