diff --git a/tests/unit/test_feature_engineering_v2.py b/tests/unit/test_feature_engineering_v2.py index 9c95aea..ad15dd1 100644 --- a/tests/unit/test_feature_engineering_v2.py +++ b/tests/unit/test_feature_engineering_v2.py @@ -10,7 +10,7 @@ def test_preprocess_v2(self): df = spark.createDataFrame([[2.8, 3.1], [0.0, 20.2]]).toDF("fixed acidity", "volatile acidity") df = df.transform(rename_columns) - actual_df = df.transform(engineer_features) + actual_df = df.transform(engineer_features) expected_df = spark.createDataFrame( [