Changed removal of whitespaces to only first row

dyvenia · Sep 26, 2024 · d521a5b · d521a5b
1 parent 2550b7b
commit d521a5b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/viadot/sources/sap_rfc.py b/src/viadot/sources/sap_rfc.py
@@ -1140,8 +1140,8 @@ def to_df(self, tests: dict | None = None) -> pd.DataFrame:  # noqa: C901, PLR09
                         # If whitespace is in unique column, it must be removed to make
                         # a proper merge.
                         for col in self.rfc_unique_id:
-                            df_tmp[col] = df_tmp[col].str.strip()
-                            df[col] = df[col].str.strip()
+                            df_tmp[col][0] = df_tmp[col][0].str.strip()
+                            df[col][0] = df[col][0].str.strip()
                         df = pd.merge(df, df_tmp, on=self.rfc_unique_id, how="outer")
                     elif not start:
                         df[fields] = records