Merge branch 'glum-v3' into convert-nas-unseen

Quantco · Jan 15, 2024 · fc6a08a · fc6a08a
2 parents c2d88b2 + 72971f4
commit fc6a08a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/glum/_glm.py b/src/glum/_glm.py
@@ -1311,6 +1311,7 @@ def linear_predictor(
             The context to use for evaluating the formula. If an integer, the
             context is taken from the stack frame of the caller at the given
             depth. If a dict, it is used as the context directly.
+
         Returns
         -------
         array, shape (n_samples, n_alphas)
@@ -1401,6 +1402,7 @@ def predict(
             The context to use for evaluating the formula. If an integer, the
             context is taken from the stack frame of the caller at the given
             depth. If a dict, it is used as the context directly.
+
         Returns
         -------
         array, shape (n_samples, n_alphas)
@@ -1474,6 +1476,7 @@ def coef_table(
             The context to use for evaluating the formula. If an integer, the
             context is taken from the stack frame of the caller at the given
             depth. If a dict, it is used as the context directly.
+
         Returns
         -------
         pandas.DataFrame
@@ -1603,6 +1606,7 @@ def wald_test(
             The context to use for evaluating the formula. If an integer, the
             context is taken from the stack frame of the caller at the given
             depth. If a dict, it is used as the context directly.
+
         Returns
         -------
         WaldTestResult
@@ -1750,6 +1754,7 @@ def _wald_test_matrix(
             If not specified, the model's ``expected_information`` attribute is used.
         context : Optional[Mapping[str, Any]], default=None
             The context to use for evaluating the formula.
+
         Returns
         -------
         WaldTestResult
@@ -1854,6 +1859,7 @@ def _wald_test_feature_names(
             If not specified, the model's ``expected_information`` attribute is used.
         context : Optional[Mapping[str, Any]], default=None
             The context to use for evaluating the formula.
+
         Returns
         -------
         WaldTestResult
@@ -1949,6 +1955,7 @@ def _wald_test_formula(
             If not specified, the model's ``expected_information`` attribute is used.
         context : Optional[Mapping[str, Any]], default=None
             The context to use for evaluating the formula.
+
         Returns
         -------
         WaldTestResult
@@ -2037,6 +2044,7 @@ def _wald_test_term_names(
             If not specified, the model's ``expected_information`` attribute is used.
         context : Optional[Mapping[str, Any]], default=None
             The context to use for evaluating the formula.
+
         Returns
         -------
         WaldTestResult
@@ -2481,6 +2489,7 @@ def score(
             The context to use for evaluating the formula. If an integer, the
             context is taken from the stack frame of the caller at the given
             depth. If a dict, it is used as the context directly.
+
         Returns
         -------
         float