Add updated description of adelie, add badge of total downloads, fix …

…a small bug in diagnostic

README.md

@@ -3,12 +3,14 @@
 </h1><br>
 
 ![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/JamesYang007/adelie/test_docs.yml)
+[![Downloads](https://static.pepy.tech/badge/adelie)](https://pepy.tech/project/adelie)
 [![PyPI Downloads](https://img.shields.io/pypi/dm/adelie.svg?label=PyPI%20downloads)](https://pypi.org/project/adelie/)
 ![versions](https://img.shields.io/pypi/pyversions/adelie.svg)
 ![PyPI - Version](https://img.shields.io/pypi/v/adelie)
 ![GitHub Release](https://img.shields.io/github/v/release/JamesYang007/adelie)
 
-Adelie is a fast and flexible Python package for solving group lasso and elastic net problems. 
+Adelie is a fast and flexible Python package for solving 
+lasso, elastic net, group lasso, and group elastic net problems. 
 
 - **Installation**: [https://jamesyang007.github.io/adelie/notebooks/installation.html](https://jamesyang007.github.io/adelie/notebooks/installation.html)
 - **Documentation**: [https://jamesyang007.github.io/adelie](https://jamesyang007.github.io/adelie/)

adelie/diagnostic.py

@@ -196,6 +196,7 @@ def objective(
     y = glm.y
 
     if groups is None:
+        p = X.shape[1]
         groups = np.arange(p, dtype=int)
 
     if glm.is_multi:

docs/sphinx/index.rst

@@ -17,7 +17,8 @@ Adelie documentation
 `Issue Tracker <https://github.com/JamesYang007/adelie/issues>`_ 
 
 
-Adelie is a fast and flexible Python package for solving group lasso and elastic net problems. 
+Adelie is a fast and flexible Python package for solving 
+lasso, elastic net, group lasso, and group elastic net problems. 
 It offers a general purpose group elastic net solver, 
 a wide range of matrix classes that can exploit special structure to allow large-scale inputs,
 and an assortment of generalized linear model (GLM) classes for fitting various types of data.