Skip to content

Commit 9147749

Browse files
mbaaksbrugman
authored andcommitted
FIX: keep gt_entity_id as int64 when no match candidates are added
1 parent 90b9885 commit 9147749

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

emm/indexing/pandas_candidate_selection.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,7 @@ def transform(self, X: pd.DataFrame) -> pd.DataFrame:
212212
if self.with_no_matches:
213213
# change gt_uid column to nullable integer
214214
candidates["gt_uid"] = candidates["gt_uid"].replace({NO_MATCH_ID: np.nan}).astype("Int64")
215+
candidates["gt_entity_id"] = candidates["gt_entity_id"].replace({NO_MATCH_ID: np.nan}).astype("Int64")
215216

216217
timer.log_param("n", len(X))
217218

0 commit comments

Comments
 (0)