list key for sequentail features

docs/source/conf.py

@@ -28,7 +28,7 @@
 project = "rialto"
 copyright = "2022, Marek Dobransky"
 author = "Marek Dobransky"
-release = "1.3.0"
+release = "2.0.1"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

rialto/maker/feature_maker.py

@@ -49,7 +49,10 @@ def _set_values(self, df: DataFrame, key: typing.Union[str, typing.List[str]], m
         :return: None
         """
         self.data_frame = df
-        self.key = key
+        if isinstance(key, str):
+            self.key = [key]
+        else:
+            self.key = key
         self.make_date = make_date
 
     def _order_by_dependencies(self, feature_holders: typing.List[FeatureHolder]) -> typing.List[FeatureHolder]:
@@ -136,7 +139,7 @@ def _make_sequential(self, keep_preexisting: bool) -> DataFrame:
             )
         if not keep_preexisting:
             logger.info("Dropping non-selected columns")
-            self.data_frame = self.data_frame.select(self.key, *feature_names)
+            self.data_frame = self.data_frame.select(*self.key, *feature_names)
         return self._filter_null_keys(self.data_frame)
 
     def _make_aggregated(self) -> DataFrame:

tests/maker/test_FeatureMaker.py

@@ -61,6 +61,13 @@ def test_sequential_multi_key(input_df):
     assert "TRANSACTIONS_OUTBOUND_VALUE" in df.columns
 
 
+def test_sequential_multi_key_drop(input_df):
+    df, _ = FeatureMaker.make(
+        input_df, ["CUSTOMER_KEY", "TYPE"], date.today(), sequential_outbound, keep_preexisting=False
+    )
+    assert "TRANSACTIONS_OUTBOUND_VALUE" in df.columns
+
+
 def test_sequential_keeps(input_df):
     df, _ = FeatureMaker.make(input_df, "CUSTOMER_KEY", date.today(), sequential_outbound, keep_preexisting=True)
     assert "AMT" in df.columns