fix: tests

quintoandar · Jun 7, 2024 · cc4fb54 · cc4fb54
1 parent e297615
commit cc4fb54
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/butterfree/pipelines/feature_set_pipeline.py b/butterfree/pipelines/feature_set_pipeline.py
@@ -136,7 +136,7 @@ def __init__(
         source: Source,
         feature_set: FeatureSet,
         sink: Sink,
-        spark_client: Optional[SparkClient],
+        spark_client: Optional[SparkClient] = None,
     ):
         self.source = source
         self.feature_set = feature_set

diff --git a/butterfree/transform/utils/function.py b/butterfree/transform/utils/function.py
@@ -34,7 +34,7 @@ def func(self, value: Callable) -> None:
         """Definitions to be used in the transformation."""
         if value is None:
             raise ValueError("Function must not be empty.")
-        if callable(value) is None:
+        if callable(value) is False:
             raise TypeError("Function must be callable.")
 
         self._func = value

diff --git a/tests/unit/butterfree/transform/transformations/test_aggregated_transform.py b/tests/unit/butterfree/transform/transformations/test_aggregated_transform.py
@@ -67,7 +67,7 @@ def test_blank_aggregation(self, feature_set_dataframe):
                 name="feature1",
                 description="unit test",
                 transformation=AggregatedTransform(
-                    functions=[Function(func="", data_type="")]
+                    functions=[Function(func=None, data_type="")]
                 ),
             )