Skip to content

Validate PyTorch repeat count dimensions#3422

Merged
FlorianPfaff merged 2 commits into
mainfrom
fix-pytorch-repeat-count-dim-validation
Jun 29, 2026
Merged

Validate PyTorch repeat count dimensions#3422
FlorianPfaff merged 2 commits into
mainfrom
fix-pytorch-repeat-count-dim-validation

Conversation

@FlorianPfaff

Copy link
Copy Markdown
Owner

Summary

  • reject nested PyTorch repeat count arrays/tensors before calling torch.repeat_interleave
  • align the PyTorch backend repeat wrapper with NumPy/PyRecEst repeat-count shape validation by raising ValueError for repeat counts deeper than 1-D
  • add regression coverage for nested NumPy-style and tensor repeat-count inputs

Bug fixed

backend.repeat(..., repeats=[[...]]) under PYRECEST_BACKEND=pytorch allowed nested repeat counts to reach Torch directly. Torch then raised a low-level RuntimeError (repeats must be 0-dim or 1-dim tensor) instead of the backend rejecting the invalid NumPy-style repeat-count shape itself.

Testing

  • Added tests/backend_support/test_pytorch_repeat_count_validation.py
  • Reproduced the contract mismatch locally with NumPy 2.3.5 and Torch 2.10.0+cpu
  • Full repository test suite was not run in this connector environment

@github-actions

Copy link
Copy Markdown
Contributor

MegaLinter analysis: Success

Descriptor Linter Files Fixed Errors Warnings Elapsed time
✅ COPYPASTE jscpd yes no no 63.57s
✅ JSON prettier 7 0 0 0 1.1s
✅ JSON v8r 7 0 0 4.65s
✅ MARKDOWN markdownlint 68 0 0 0 1.81s
✅ MARKDOWN markdown-table-formatter 68 0 0 0 0.81s
✅ PYTHON black 1122 10 0 0 53.46s
✅ PYTHON isort 1122 15 0 0 2.88s
✅ REPOSITORY checkov yes no no 44.62s
✅ REPOSITORY gitleaks yes no no 11.52s
✅ REPOSITORY git_diff yes no no 0.12s
✅ REPOSITORY secretlint yes no no 33.96s
✅ REPOSITORY syft yes no no 2.74s
✅ REPOSITORY trivy-sbom yes no no 11.1s
✅ REPOSITORY trufflehog yes no no 16.54s
✅ YAML prettier 11 0 0 0 0.71s
✅ YAML v8r 11 0 0 10.39s
✅ YAML yamllint 11 0 0 0.47s

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 merged commit 1adebfc into main Jun 29, 2026
14 of 24 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.

1 participant