From f9406df0fb2a07bc99ecca640d9113cf540465d1 Mon Sep 17 00:00:00 2001
From: Scott Linderman <scott.linderman@gmail.com>
Date: Tue, 17 Oct 2017 17:13:25 -0400
Subject: [PATCH] adding robust distributions to list of all distributions
---
examples/robust_regression.py | 2 +-
pybasicbayes/distributions/regression.py | 3 ++-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/examples/robust_regression.py b/examples/robust_regression.py
index 16139b6..8ba563e 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 ef2b0f5..2ca302a 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
--
GitLab