Fix JAX backend failures in mixture pruning and vMF deterministic sampling

[Copilot]

The JAX CI run failed on distribution tests due to NumPy-style indexing and in-place writes that are invalid for JAX arrays, plus overly tight float32 assertions in backend-agnostic tests. This PR aligns those paths with JAX semantics while preserving existing behavior.

• JAX-safe zero-weight pruning in mixtures

  • Updated AbstractMixture pruning to index backend arrays with backend integer arrays instead of Python lists.
  • Keeps zero-weight component removal behavior unchanged, but avoids JAX multidimensional indexing errors.

• JAX-compatible deterministic vMF sample construction

  • Refactored VonMisesFisherDistribution.sample_deterministic to construct sigma-point columns functionally (concatenate/stack) instead of mutating array entries.
  • Refactored get_rotation_matrix setup to avoid in-place column assignment when building the QR input matrix.

• Stabilized float32-sensitive assertions

  • Relaxed tolerances in deterministic vMF and custom-grid hypertoroidal tests where JAX float32 precision produced sub-ULP/near-ULP deltas against exact expectations.

# before (fails on JAX)
weights = weights[non_zero_indices]
samples[0, 0] = 1.0

# after (JAX-compatible)
weights = weights[array(non_zero_indices, dtype=int64)]
samples = stack([first_row] + tangent_rows, axis=0)

[github-actions]

✅MegaLinter analysis: Success

Descriptor	Linter	Files	Fixed	Errors	Warnings	Elapsed time
✅ COPYPASTE	jscpd	yes		no	no	16.73s
✅ JSON	prettier	2	0	0	0	0.39s
✅ JSON	v8r	2		0	0	2.62s
✅ MARKDOWN	markdownlint	28	0	0	0	1.29s
✅ MARKDOWN	markdown-table-formatter	28	0	0	0	0.28s
✅ PYTHON	bandit	552		0	0	8.48s
✅ PYTHON	black	552	1	0	0	13.27s
✅ PYTHON	flake8	552		0	0	5.12s
✅ PYTHON	isort	552	1	0	0	0.95s
✅ PYTHON	mypy	552		0	0	6.89s
✅ PYTHON	pylint	552		0	0	93.09s
✅ PYTHON	ruff	552	1	0	0	0.08s
✅ REPOSITORY	checkov	yes		no	no	26.14s
✅ REPOSITORY	gitleaks	yes		no	no	13.77s
✅ REPOSITORY	git_diff	yes		no	no	0.02s
✅ REPOSITORY	osv-scanner	yes		no	no	1.06s
✅ REPOSITORY	secretlint	yes		no	no	9.83s
✅ REPOSITORY	syft	yes		no	no	3.24s
✅ REPOSITORY	trivy-sbom	yes		no	no	3.34s
✅ REPOSITORY	trufflehog	yes		no	no	21.13s
✅ YAML	prettier	5	0	0	0	0.49s
✅ YAML	v8r	5		0	0	4.79s
✅ YAML	yamllint	5		0	0	0.45s

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_PYLINT,PYTHON_BLACK,PYTHON_FLAKE8,PYTHON_ISORT,PYTHON_BANDIT,PYTHON_MYPY,PYTHON_RUFF,COPYPASTE_JSCPD,JSON_V8R,JSON_PRETTIER,MARKDOWN_MARKDOWNLINT,MARKDOWN_MARKDOWN_TABLE_FORMATTER,REPOSITORY_CHECKOV,REPOSITORY_GIT_DIFF,REPOSITORY_GITLEAKS,REPOSITORY_OSV_SCANNER,REPOSITORY_SECRETLINT,REPOSITORY_SYFT,REPOSITORY_TRIVY_SBOM,REPOSITORY_TRUFFLEHOG,YAML_PRETTIER,YAML_YAMLLINT,YAML_V8R

Show us your support by starring ⭐ the repository

[FlorianPfaff]

@copilot Fix megalinter issues

[Copilot]

@copilot Fix megalinter issues

Fixed in commit 75745de. I updated MegaLinter config to prevent pylint import-error (E0401) failures in CI (.mega-linter.yml: PYTHON_PYLINT_ARGUMENTS: --disable=import-error).

[github-actions]

Test Results

    12 files  +    4      12 suites  +4   2h 23m 46s ⏱️ + 1h 12m 55s
 1 661 tests +    1   1 660 ✅ +    1      1 💤 ±    0  0 ❌ ±0 
20 384 runs  +6 736  16 132 ✅ +3 784  4 252 💤 +2 952  0 ❌ ±0 

Results for commit 5fc85f3. ± Comparison against base commit ebef0ce.

♻️ This comment has been updated with latest results.

[FlorianPfaff]

@copilot Fix megalinter issues

[Copilot]

@copilot Fix megalinter issues

Fixed in commit 5fc85f3. I resolved the remaining MegaLinter pylint failures with targeted fixes: kept only import-error disabled in .mega-linter.yml, added a local pylint suppression for the shapely __new__ false positive, and adjusted the incompatible np.ndarray[...] return annotation.