Skip to content

fix: allow num_labels property to return None when id2label is unset #47069

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
SebTardif wants to merge 2 commits into
base: main
Choose a base branch
from
Open

fix: allow num_labels property to return None when id2label is unset #47069

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
bed3341
fix: allow num_labels property to return None when id2label is unset
SebTardif Jul 4, 2026
9a87e2d
Merge branch 'main' into fix/num-labels-optional-type
zucchini-nlp Jul 6, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions src/transformers/configuration_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -351,9 +351,10 @@ def name_or_path(self, value):
self._name_or_path = str(value) # Make sure that name_or_path is a string (for JSON encoding)

@property
def num_labels(self) -> int:
def num_labels(self) -> int | None:
"""
`int`: The number of labels for classification models.
`int` or `None`: The number of labels for classification models, or
`None` when `id2label` is not set.
"""
return len(self.id2label) if self.id2label is not None else None

Expand Down
Loading