Fix test failures due to old change_flags API

[maresb]

Breakage was caused by this:
pymc-devs/pytensor@936554c Fix is to unpack the dict into kwargs.

Description

Related Issue

• Closes #
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

📚 Documentation preview 📚: https://pymc--7475.org.readthedocs.build/en/7475/

[ricardoV94]

The other failing test may be something that @jessegrabowski knows about

[maresb]

Is there an issue or PR tracking this other failing test?

AILED tests/distributions/test_multivariate.py::TestMatchesScipy::test_kroneckernormal[None-3-2] - ValueError: shape mismatch: value array of shape (2,) could not be broadcast to indexing result of shape (3,) Apply node that caused the error: AdvancedSetSubtensor(Alloc.0, *0-<Vector(float64, shape=(3,))>, [0 1 2], [0 1 2]) Toposort index: 1 Inputs types: [TensorType(float64, shape=(3, 3)), TensorType(float64, shape=(3,)), TensorType(uint8, shape=(3,)), TensorType(uint8, shape=(3,))] Inputs shapes: [(3, 3), (2,), (3,), (3,)] Inputs strides: [(24, 8), (8,), (1,), (1,)] Inputs values: ['not shown', array([2.00033806, 3.69369647]), array([0, 1, 2], dtype=uint8), array([0, 1, 2], dtype=uint8)] Inputs type_num: [12, 12, 2, 2] Outputs clients: [[output[0](AdvancedSetSubtensor.0)]] Backtrace when the node is created (use PyTensor flag traceback__limit=N to make it longer): File "/usr/share/miniconda3/envs/pymc-test/lib/python3.12/site-packages/pluggy/_hooks.py", line 513, in __call__ return self._hookexec(self.name, self._hookimpls.copy(), kwargs, firstresult) File "/usr/share/miniconda3/envs/pymc-test/lib/python3.12/site-packages/pluggy/_manager.py", line 120, in _hookexec return self._inner_hookexec(hook_name, methods, kwargs, firstresult) File "/usr/share/miniconda3/envs/pymc-test/lib/python3.12/site-packages/pluggy/_callers.py", line 103, in _multicall res = hook_impl.function(*args) File "/usr/share/miniconda3/envs/pymc-test/lib/python3.12/site-packages/_pytest/python.py", line 159, in pytest_pyfunc_call result = testfunction(**testargs) File "/home/runner/work/pymc/pymc/tests/distributions/test_multivariate.py", line 460, in test_kroneckernormal check_logp( File "/home/runner/work/pymc/pymc/pymc/testing.py", line 356, in check_logp dist = create_dist_from_paramdomains(pymc_dist, paramdomains, extra_args) File "/home/runner/work/pymc/pymc/pymc/testing.py", line 271, in create_dist_from_paramdomains return pymc_dist.dist(**param_vars, **extra_args) File "/home/runner/work/pymc/pymc/pymc/distributions/multivariate.py", line 2087, in dist cov_i = pt.dot(Q, pt.dot(pt.diag(eig), Q.T)) Debug print of the apply node: AdvancedSetSubtensor [id A] <Matrix(float64, shape=(?, ?))> ├─ Alloc [id B] <Matrix(float64, shape=(3, 3))> │ ├─ 0.0 [id C] <Scalar(float64, shape=())> │ ├─ 3 [id D] <Scalar(int64, shape=())> │ └─ 3 [id D] <Scalar(int64, shape=())> ├─ *0-<Vector(float64, shape=(3,))> [id E] <Vector(float64, shape=(3,))> ├─ [0 1 2] [id F] <Vector(uint8, shape=(3,))> └─ [0 1 2] [id F] <Vector(uint8, shape=(3,))> Storage map footprint: - Alloc.0, Shape: (3, 3), ElemSize: 8 Byte(s), TotalSize: 72 Byte(s) - AdvancedSetSubtensor.0, Shape: (3, 3), ElemSize: 8 Byte(s), TotalSize: 72 Byte(s) - *0-<Vector(float64, shape=(3,))>, Input, Shape: (2,), ElemSize: 8 Byte(s), TotalSize: 16 Byte(s) - 0.0, Shape: (), ElemSize: 8 Byte(s), TotalSize: 8.0 Byte(s) - 3, Shape: (), ElemSize: 8 Byte(s), TotalSize: 8.0 Byte(s) - [0 1 2], Shape: (3,), ElemSize: 1 Byte(s), TotalSize: 3 Byte(s) TotalSize: 107.0 Byte(s) 0.000 GB TotalSize inputs: 35.0 Byte(s) 0.000 GB

[ricardoV94]

It will be solved by pymc-devs/pytensor#981