X, y, _ = _make_data()
alpha = .1
mask = np.ones(X.shape[1]).astype(np.bool)
// results should be exactly the same for pure lasso
a = tvl1_solver(X, y, alpha, 1., mask, loss="mse", max_iter=max_iter)[0]
After Change
X, y, _, mask = _make_data()
alpha = .1
unmasked_X = np.rollaxis(X, -1, start=0)
unmasked_X = np.array([x[mask] for x in unmasked_X])
// results should be exactly the same for pure lasso
a = tvl1_solver(unmasked_X, y, alpha, 1., mask, loss="mse",