Reassign dataframe after dropping extra columns

ml6team · Jun 23, 2023 · 4442188 · 4442188
1 parent 309a886
commit 4442188
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fondant/component.py b/fondant/component.py
@@ -273,7 +273,7 @@ def wrapped_transform(self, dataframe: pd.DataFrame) -> pd.DataFrame:
         )
         dataframe = self.transform(dataframe)
         # Drop columns not in the produces section of the component spec
-        dataframe.drop(
+        dataframe = dataframe.drop(
             columns=[
                 (subset, field)
                 for (subset, field) in dataframe.columns