diff --git a/src/_py/utils.rs b/src/_py/utils.rs index 1799f1a..45c24f2 100644 --- a/src/_py/utils.rs +++ b/src/_py/utils.rs @@ -5,18 +5,31 @@ use pyo3_polars::{PyDataFrame, PySeries}; #[allow(deprecated)] #[pymodule] pub fn utils(_py: Python, m: &PyModule) -> PyResult<()> { - m.add_function(wrap_pyfunction!(get_bdates_series_default, m)?)?; + m.add_function(wrap_pyfunction!(get_bdates_series_default_pl, m)?)?; + m.add_function(wrap_pyfunction!(get_bdates_series_default_opt, m)?)?; Ok(()) } #[pyfunction] -pub fn get_bdates_series_default( +pub fn get_bdates_series_default_pl( start_date: String, end_date: String, freq: Option, ) -> PyResult { Ok(PySeries( - crate::utils::dateutils::get_bdates_series_default(start_date, end_date, freq) + crate::utils::dateutils::get_bdates_series_default_pl(start_date, end_date, freq) + .map_err(|e| PyErr::new::(format!("{}", e)))?, + )) +} + +#[pyfunction] +pub fn get_bdates_series_default_opt( + start_date: String, + end_date: String, + freq: Option, +) -> PyResult { + Ok(PySeries( + crate::utils::dateutils::get_bdates_series_default_opt(start_date, end_date, freq) .map_err(|e| PyErr::new::(format!("{}", e)))?, )) }