fix bayes.py import