From 287daec04f2094f47f8c3c3d3e2b131f0068839f Mon Sep 17 00:00:00 2001 From: mvansegbroeck Date: Wed, 20 Sep 2023 22:02:54 -0700 Subject: [PATCH] Update ancestry.py --- src/gretel_trainer/relational/ancestry.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gretel_trainer/relational/ancestry.py b/src/gretel_trainer/relational/ancestry.py index 6dc23860..db2a64d1 100644 --- a/src/gretel_trainer/relational/ancestry.py +++ b/src/gretel_trainer/relational/ancestry.py @@ -162,7 +162,7 @@ def _join_parents( left_on_cols = [f"{lineage}{_END_LINEAGE}{col}" for col in foreign_key.columns] right_on_cols = [f"{next_lineage}{_END_LINEAGE}{parent_col}" for parent_col in foreign_key.parent_columns] for left_col, right_col in zip(left_on_cols, right_on_cols): - if parent_data[right_col].dtype == "object" + if parent_data[right_col].dtype == "object": df[left_col] = df[left_col].astype(parent_data[right_col].dtype) df = df.merge(parent_data, how="left", left_on=left_on_cols, right_on=right_on_cols)