We currently don't have any type hinting for any of the functions in simulate_data.py. It would be good to add this. That will also act as an improvement to the auto generated API docs.