What's Changed
* Test with latest PyMC by michaelosthege in https://github.com/michaelosthege/pytensor-federated/pull/63
* Fix return type hints by michaelosthege in https://github.com/michaelosthege/pytensor-federated/pull/64
**Full Changelog**: https://github.com/michaelosthege/pytensor-federated/compare/v1.0.1...v1.0.2