Reassign dataframe after dropping extra columns (#233)

Small bug introduced by refactoring after testing #226
ml6team · Jun 26, 2023 · 6ef449c · 6ef449c
1 parent 309a886
commit 6ef449c
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