fix _apply_transformation_function

logicalclocks · Jul 9, 2024 · 16e40f7 · 16e40f7
1 parent 5099d5d
commit 16e40f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/hsfs/engine/spark.py b/python/hsfs/engine/spark.py
@@ -1199,7 +1199,7 @@ def add_cols_to_delta_table(self, feature_group, new_features):
             "spark.databricks.delta.schema.autoMerge.enabled", "true"
         ).save(feature_group.location)
 
-    def _apply_transformation_function(self, transformation_functions, dataset):
+    def _apply_transformation_function(self, transformation_functions, dataset, **kwargs):
         # generate transformation function expressions
         transformed_feature_names = []
         transformation_fn_expressions = []