From ee531deb7a991c0030860c16847ae32b8441ad99 Mon Sep 17 00:00:00 2001
From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com>
Date: Wed, 16 Apr 2025 19:27:46 +0100
Subject: [PATCH] update create_blacklist_from_qdf function signature to accept
 metrics as a vector

---
 src/_py/utils.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/_py/utils.rs b/src/_py/utils.rs
index 5584578..e3b6e65 100644
--- a/src/_py/utils.rs
+++ b/src/_py/utils.rs
@@ -1,7 +1,6 @@
 use pyo3::{prelude::*, types::PyDict};
 use pyo3_polars::{PyDataFrame, PySeries};
 
-
 /// Python wrapper for [`crate::utils::qdf`] module.
 #[allow(deprecated)]
 #[pymodule]
@@ -37,18 +36,18 @@ pub fn get_bdates_series_default_opt(
 }
 
 #[allow(deprecated)]
-#[pyfunction(signature = (df, group_by_cid=None, blacklist_name=None, metric=None))]
+#[pyfunction(signature = (df, group_by_cid=None, blacklist_name=None, metrics=None))]
 pub fn create_blacklist_from_qdf(
     df: PyDataFrame,
     group_by_cid: Option<bool>,
     blacklist_name: Option<String>,
-    metric: Option<String>,
+    metrics: Option<Vec<String>>,
 ) -> PyResult<PyObject> {
     let result = crate::utils::qdf::blacklist::create_blacklist_from_qdf(
         &df.into(),
         group_by_cid,
         blacklist_name,
-        metric,
+        metrics,
     )
     .map_err(|e| PyErr::new::<pyo3::exceptions::PyValueError, _>(format!("{}", e)))?;
     Python::with_gil(|py| {