Skip to content

Fix PyTorch one_hot scalar labels#3643

Open
FlorianPfaff wants to merge 3 commits into
mainfrom
fix-pytorch-one-hot-scalar-20260701
Open

Fix PyTorch one_hot scalar labels#3643
FlorianPfaff wants to merge 3 commits into
mainfrom
fix-pytorch-one-hot-scalar-20260701

Conversation

@FlorianPfaff

Copy link
Copy Markdown
Owner

Summary

  • Add a PyTorch one_hot compatibility hook that treats scalar integer labels as label values instead of tensor sizes.
  • Keep public and raw PyTorch backend one_hot synchronized after package import.
  • Add PyTorch-specific regression coverage for scalar-label one-hot encoding.

Bug fixed

The PyTorch backend converted non-tensor labels with torch.LongTensor(labels). For scalar integer labels such as 1, PyTorch interprets the integer as a tensor length and allocates an uninitialized label tensor, which can raise out-of-range errors or produce nondeterministic results. The new hook uses scalar-index normalization for scalar labels while preserving the existing sequence-label behavior.

Testing

  • Not run in this GitHub-only session; focused regression tests were added.

@github-actions

github-actions Bot commented Jul 1, 2026

Copy link
Copy Markdown
Contributor

MegaLinter analysis: Success

Descriptor Linter Files Fixed Errors Warnings Elapsed time
✅ COPYPASTE jscpd yes no no 79.73s
✅ JSON prettier 7 0 0 0 0.83s
✅ JSON v8r 7 0 0 4.74s
✅ MARKDOWN markdownlint 68 0 0 0 2.07s
✅ MARKDOWN markdown-table-formatter 68 0 0 0 1.0s
✅ PYTHON black 1187 29 0 0 69.77s
✅ PYTHON isort 1187 53 0 0 7.28s
✅ REPOSITORY checkov yes no no 48.59s
✅ REPOSITORY gitleaks yes no no 12.57s
✅ REPOSITORY git_diff yes no no 0.21s
✅ REPOSITORY secretlint yes no no 45.78s
✅ REPOSITORY syft yes no no 4.31s
✅ REPOSITORY trivy-sbom yes no no 7.45s
✅ REPOSITORY trufflehog yes no no 21.26s
✅ YAML prettier 11 0 0 0 0.77s
✅ YAML v8r 11 0 0 12.16s
✅ YAML yamllint 11 0 0 0.55s

Notices

📣 MegaLinter 9.5.0 is out! Discover the new features and security recommendations in the release announcement. (Skip this info by defining SECURITY_SUGGESTIONS: false)

See detailed reports in MegaLinter artifacts

Your project could benefit from a custom flavor, which would allow you to run only the linters you need, and thus improve runtime performances. (Skip this info by defining FLAVOR_SUGGESTIONS: false)

  • Documentation: Custom Flavors
  • Command: npx mega-linter-runner@9.5.0 --custom-flavor-setup --custom-flavor-linters PYTHON_BLACK,PYTHON_ISORT,COPYPASTE_JSCPD,JSON_V8R,JSON_PRETTIER,MARKDOWN_MARKDOWNLINT,MARKDOWN_MARKDOWN_TABLE_FORMATTER,REPOSITORY_CHECKOV,REPOSITORY_GIT_DIFF,REPOSITORY_GITLEAKS,REPOSITORY_SECRETLINT,REPOSITORY_SYFT,REPOSITORY_TRIVY_SBOM,REPOSITORY_TRUFFLEHOG,YAML_PRETTIER,YAML_YAMLLINT,YAML_V8R

MegaLinter is graciously provided by OX Security
Show us your support by starring ⭐ the repository

@FlorianPfaff FlorianPfaff enabled auto-merge (squash) July 1, 2026 23:19
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