add data type to filter plugin output values

CountESS-Project · Jul 10, 2024 · e6faae9 · e6faae9
1 parent 843c5e9
commit e6faae9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/countess/plugins/filter.py b/countess/plugins/filter.py
@@ -9,6 +9,7 @@
     BooleanParam,
     ChoiceParam,
     ColumnChoiceParam,
+    DataTypeChoiceParam,
     MultiParam,
     StringParam,
     TabularMultiParam,
@@ -56,6 +57,7 @@ class FilterPlugin(PandasSimplePlugin):
                             {
                                 "output": StringParam("Output Column"),
                                 "value": StringParam("Output Value"),
+                                "type": DataTypeChoiceParam("Output Type", "string"),
                             },
                         ),
                     ),
@@ -71,7 +73,7 @@ def process(self, data: pd.DataFrame, source: str, logger: Logger) -> Iterable[p
 
         for filt in self.parameters["filters"]:
             # build a dictionary of columns to assign to matched rows.
-            assign_dict = {p["output"].value: p["value"].value for p in filt["outputs"]}
+            assign_dict = {p["output"].value: p["type"].cast_value(p["value"].value) for p in filt["outputs"]}
 
             if not filt.columns.params:
                 # If there are no filter columns at all, then we match