[MAINT/TST] add generate_lasso to strict linting

statsmodels · May 3, 2019 · 5b5fb0b · 5b5fb0b
1 parent a329388
commit 5b5fb0b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lint.sh b/lint.sh
@@ -32,6 +32,7 @@ if [ "$LINT" == true ]; then
         statsmodels/regression/recursive_ls.py \
         statsmodels/regression/tests/test_lme.py \
         statsmodels/regression/tests/test_quantile_regression.py \
+        statsmodels/regression/tests/results/generate_lasso.py \
         statsmodels/regression/tests/results/generate_lme.py \
         statsmodels/stats/__init__.py \
         statsmodels/stats/_knockoff.py \

diff --git a/statsmodels/regression/tests/results/generate_lasso.py b/statsmodels/regression/tests/results/generate_lasso.py
@@ -15,12 +15,12 @@
 
 np.random.seed(83423)
 
-exog = np.random.normal(size=(n,p))
+exog = np.random.normal(size=(n, p))
 params = (-1.)**np.arange(p)
 params[::3] = 0
 expval = np.dot(exog, params)
 endog = expval + np.random.normal(size=n)
-data = np.concatenate((endog[:,None], exog), axis=1)
+data = np.concatenate((endog[:, None], exog), axis=1)
 data = np.around(100*data)
 
 fname = "lasso_data.csv"