pymdp: A Python library for active inference in discrete state spaces