Skip to content

Fix: Use configured TransformerDeID field names - #1224

Open
fbonc wants to merge 1 commit into
sunlabuiuc:masterfrom
fbonc:fix-transformer-deid-field-names
Open

Fix: Use configured TransformerDeID field names#1224
fbonc wants to merge 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.
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.

1 participant