Update ancestry.py

gretelai · Mar 6, 2024 · 287daec · 287daec
1 parent 910d175
commit 287daec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)