Skip to content

Handle scalar labels in the data collator - #48

Merged
BioMikeUkr merged 1 commit into
Knowledgator:mainfrom
dajiaohuang:fix/scalar-label-collation
Sep 24, 2026
Merged

BioMikeUkr merged 1 commit into
Knowledgator:mainfrom
dajiaohuang:fix/scalar-label-collation

Conversation

@dajiaohuang

@dajiaohuang dajiaohuang commented Sep 23, 2026 •

Copy link
Copy Markdown
Contributor

Summary

  • Stack zero-dimensional tensor labels into a batch tensor.
  • Add a regression test for scalar-label classification.

Fixes #31.

Validation: the focused test suite passed (11 tests), the full suite passed (149 tests), and Ruff checks plus git diff check passed.

@BioMikeUkr

Copy link
Copy Markdown
Member

Looks good

@BioMikeUkr
BioMikeUkr merged commit af37666 into Knowledgator:main Sep 24, 2026
3 of 4 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.

Skipping iteration due to error: 'list' object has no attribute 'dim'

2 participants