The helper `apply_local_dimshuffle_lift` requires a FunctionGraph when elemwise inputs are involved.

Canonical slice start and stop values depend on the sign of the step. The rewrite wrongly assumed they were always 0:len(dim)

This is a special-edge case that behaves differently than other boolean indexing. It adds a new dimension (either empty or with one entry). The logic in `make_node` and `infer_shape` don't handle this.

updates:
- [github.com/astral-sh/ruff-pre-commit: v0.7.0 → v0.7.1](https://github.com/astral-sh/ruff-pre-commit/compare/v0.7.0...v0.7.1)

updates:
- [github.com/astral-sh/ruff-pre-commit: v0.6.9 → v0.7.0](https://github.com/astral-sh/ruff-pre-commit/compare/v0.6.9...v0.7.0)

* Blockwise optimal linear control ops

* Add jax rewrite to eliminate `BilinearSolveDiscreteLyapunov`

* set `solve_discrete_lyapunov` method default to bilinear

* Appease mypy

* restore method dispatching

* Use `pt.vectorize` on base `solve_discrete_lyapunov` case

* Apply JAX rewrite before canonicalization

* Improve tests

* Remove useless warning filters

* Fix local_blockwise_alloc rewrite

The rewrite was squeezing too many dimensions of the alloced value, when this didn't have dummy expand dims to the left.

* Fix float32 tests

* Test against complex inputs

* Appease ViPy (Vieira-py type checking)

* Remove condition from `TensorLike` import

* Infer dtype from `node.outputs.type.dtype`

* Remove unused mypy ignore

* Don't manually set dtype of output

Revert change to `_solve_discrete_lyapunov`

* Set dtype of Op outputs

---------
Co-authored-by: ricardoV94 <ricardo.vieira1994@gmail.com>

updates:
- [github.com/pre-commit/pre-commit-hooks: v4.6.0 → v5.0.0](https://github.com/pre-commit/pre-commit-hooks/compare/v4.6.0...v5.0.0)
- [github.com/astral-sh/ruff-pre-commit: v0.6.5 → v0.6.9](https://github.com/astral-sh/ruff-pre-commit/compare/v0.6.5...v0.6.9)

Co-authored-by: Ricardo Vieira <28983449+ricardov94@users.noreply.github.com>

Co-authored-by: Ricardo Vieira <28983449+ricardov94@users.noreply.github.com>

Only run required rewrites in JAX tests

Several tests ended up not testing the backend Op implementations due to constant folding of inputs.

Bumps [pypa/gh-action-pypi-publish](https://github.com/pypa/gh-action-pypi-publish) from 1.9.0 to 1.10.3.
- [Release notes](https://github.com/pypa/gh-action-pypi-publish/releases)
- [Commits](https://github.com/pypa/gh-action-pypi-publish/compare/v1.9.0...v1.10.3)

---
updated-dependencies:
- dependency-name: pypa/gh-action-pypi-publish
  dependency-type: direct:production
  update-type: version-update:semver-minor
...
Signed-off-by: dependabot[bot] <support@github.com>

* fixed merge conflicts

* fixed failing tests and added rewrite for pt.diag

* minor changes; added test to not apply rewrite

* added test for batched case and more cases of not applying rewrite

* minor changes

* Added rewrite for diag of kronecker product

* Added rewrite for slogdet; added docstrings for  rewrites

* fixed typo

* added rewrite for diag(block_diag)

* added rewrite for determinant of blockdiag

* Added rewrite for slogdet; added docstrings for all 3 rewrites

* fixed typecasting for tests