Skip to content

Fix: Use configured TransformerDeID field names - #1224

Merged
lehendo merged 1 commit into
sunlabuiuc:masterfrom
fbonc:fix-transformer-deid-field-names
Sep 3, 2026
Merged

lehendo merged 1 commit into
sunlabuiuc:masterfrom
fbonc:fix-transformer-deid-field-names

Conversation

@fbonc

@fbonc fbonc commented Sep 1, 2026

Copy link
Copy Markdown
Contributor

Issue
TransformerDeID.deidentify() always called the model with text= and labels=, even when the dataset configured different feature and label field names. Custom schemas therefore failed with a missing-key error

Fix
Build the de-identification call from self.feature_key and self.label_key so inference follows the field names established from the dataset schema

Notes
Regression test added

Route de-identification inputs through the model's configured feature and label keys.

@lehendo lehendo left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me!

@lehendo
lehendo merged commit 89bb9ea into sunlabuiuc:master Sep 3, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants