Equvilant APIs in PyMC v4

This solved my issue

Thank you @ricardoV94 !