diff --git a/examples/robust_regression.py b/examples/robust_regression.py index 16139b6a886f88fb6cd6d86b9b685633c31f6e12..8ba563e4261ba03967407888a2e0b744769b6333 100644 --- a/examples/robust_regression.py +++ b/examples/robust_regression.py @@ -9,7 +9,7 @@ import seaborn as sns sns.set_style("white") from pybasicbayes.util.text import progprint_xrange -from pybasicbayes.distributions.regression import Regression, RobustRegression +from pybasicbayes.distributions import Regression, RobustRegression D_out = 1 D_in = 2 diff --git a/pybasicbayes/distributions/regression.py b/pybasicbayes/distributions/regression.py index ef2b0f5ae410c08e3a4a62cfcf2ca18b70e09955..2ca302a46518153c30d55710c7a26a886bc055e3 100644 --- a/pybasicbayes/distributions/regression.py +++ b/pybasicbayes/distributions/regression.py @@ -2,7 +2,8 @@ from __future__ import division from builtins import zip from builtins import range __all__ = ['Regression', 'RegressionNonconj', 'ARDRegression', - 'AutoRegression', 'ARDAutoRegression', 'DiagonalRegression'] + 'AutoRegression', 'ARDAutoRegression', 'DiagonalRegression', + 'RobustRegression', 'RobustAutoRegression'] import numpy as np from numpy import newaxis as na