In that script CustomDist was found. Where are you seeing the problem, perhaps in a jupyter notebook? In that case it may not be linked to the right environment (kernel).
When I asked you to print pymc.__version__ and pymc.__path__, I meant you should do it in the same environment/script/notebook where you were getting the error