From 658eb0d1214a0421bfb3e927571f301b2668ed93 Mon Sep 17 00:00:00 2001 From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com> Date: Sun, 13 Apr 2025 11:10:14 +0100 Subject: [PATCH] Rename and update functions in utils module for clarity: change get_bdates_series_default to get_bdates_series_default_pl --- src/_py/utils.rs | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) 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)))?, )) }