From 89b86091f9c4a7786ba120957e6fd9b4f437a8e3 Mon Sep 17 00:00:00 2001 From: Lorenzo Baldacci <61754353+l-baldacci@users.noreply.github.com> Date: Wed, 13 Jul 2022 17:47:28 +0100 Subject: [PATCH] Update test_feature_engineering_v2.py --- tests/unit/test_feature_engineering_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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( [