Skip to content

Fix raw JAX std out contract#3444

Open
FlorianPfaff wants to merge 2 commits into
mainfrom
fix-raw-jax-std-out-default-backend-v2
Open

Fix raw JAX std out contract#3444
FlorianPfaff wants to merge 2 commits into
mainfrom
fix-raw-jax-std-out-default-backend-v2

Conversation

@FlorianPfaff

Copy link
Copy Markdown
Owner

Summary

  • patch raw pyrecest._backend.jax.std from backend-support initialization so out= is handled even when the public backend is not JAX
  • keep the public JAX facade aligned when JAX is the active backend
  • add a subprocess regression for PYRECEST_BACKEND=numpy followed by direct raw JAX std(..., out=...)

Bug fixed

The existing JAX std(..., out=...) compatibility patch only ran through the active public JAX facade path. With PYRECEST_BACKEND=numpy, importing pyrecest and then using pyrecest._backend.jax.std(values, out=...) could still expose jax.numpy.std, which rejects non-None out values. The raw backend now computes with out=None and returns the JAX .at[...] updated output array.

Testing

  • Not run locally in this connector session; added focused regression coverage in tests/backend_support/test_raw_jax_std_out_default_backend.py.

@FlorianPfaff FlorianPfaff enabled auto-merge (squash) June 29, 2026 15:40
@github-actions

Copy link
Copy Markdown
Contributor

MegaLinter analysis: Success

Descriptor Linter Files Fixed Errors Warnings Elapsed time
✅ COPYPASTE jscpd yes no no 77.15s
✅ JSON prettier 7 0 0 0 1.04s
✅ JSON v8r 7 0 0 4.65s
✅ MARKDOWN markdownlint 68 0 0 0 2.09s
✅ MARKDOWN markdown-table-formatter 68 0 0 0 0.87s
✅ PYTHON black 1132 4 0 0 64.47s
✅ PYTHON isort 1132 4 0 0 5.47s
✅ REPOSITORY checkov yes no no 50.06s
✅ REPOSITORY gitleaks yes no no 9.02s
✅ REPOSITORY git_diff yes no no 0.16s
✅ REPOSITORY secretlint yes no no 33.9s
✅ REPOSITORY syft yes no no 5.16s
✅ REPOSITORY trivy-sbom yes no no 12.49s
✅ REPOSITORY trufflehog yes no no 20.8s
✅ YAML prettier 11 0 0 0 0.75s
✅ YAML v8r 11 0 0 11.61s
✅ YAML yamllint 11 0 0 0.62s

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 force-pushed the fix-raw-jax-std-out-default-backend-v2 branch from 0110238 to bb32708 Compare June 29, 2026 19:03
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