Fixed docs build

Author: jmafoster1

causal_testing/discovery/nsga_discovery.py

@@ -39,7 +39,7 @@ def binary_string_to_causal_dag(self, individual: np.array) -> CausalDAG:
         Converts a binary string representation of a causal DAG to a CausalDAG object.
 
         :param individual: Bitstring of the same length as `possible_edges` such that 1 at position `i` represents
-        possible_edges[i] being an edge in the graph and 0 represents it not being.
+                           possible_edges[i] being an edge in the graph and 0 represents it not being.
         :returns: The converted CausalDAG instance.
         """
         causal_dag = CausalDAG()

causal_testing/estimation/abstract_regression_estimator.py

@@ -64,6 +64,7 @@ def __init__(
     def _get_adjusted_variables(self, tree: ast.AST) -> set[str]:
         """
         Recursively return variables in an AST.
+
         :returns: Set of all variables not used as part of a function.
         """
         if isinstance(tree, ast.Name) and tree.id != self.treatment_variable:
@@ -108,7 +109,9 @@ def _setup_covariates(self, df: pd.DataFrame) -> pd.Series:
         Parse the formula and set up the covariates from the design matrix so we can use them in the statsmodels array
         API. This allows us to only parse the formula once, rather than using the formula API, which parses it every
         time the regression model is fit, which can be a lot if using causal test adequacy.
+
         :param df: The data to use.
+
         :returns: The data and the covariate columns.
         """
         _, covariate_data = dmatrices(self.formula, df, return_type="dataframe")
@@ -154,9 +157,10 @@ def treatment_columns(self, model: RegressionResultsWrapper) -> list[str]:
         This is a workaround for statsmodels mangling the names of categorical variables to include the values.
 
         :param model: The fitted model from which to extract the variable names.
+
         :returns: A list of the feature names in the model that represent the treatment. Normally this will just be
-        [treatment_name], but for categorical treatments, you'll have
-        [treatment_name[value_1], treatment_name[value_2]].
+                  [treatment_name], but for categorical treatments, you'll have
+                  [treatment_name[value_1], treatment_name[value_2]].
         """
         return [
             param
@@ -170,7 +174,7 @@ def _predict(self, df) -> pd.DataFrame:
         :param df: The data to use.
         :param: adjustment_config: The values of the adjustment variables to use.
 
-        :return: The estimated outcome under control and treatment, with confidence intervals in the form of a
+        :returns: The estimated outcome under control and treatment, with confidence intervals in the form of a
                  dataframe with columns "predicted", "se", "ci_lower", and "ci_upper".
         """
         model = self.fit_model(df)

causal_testing/testing/causal_test_case.py

@@ -20,6 +20,7 @@ class CausalTestCase:
     variables, a CausalTestCase stores the values of these variables. Also the outcome variable and value are
     specified. The goal of a CausalTestCase is to test whether the intervention made to the control via the treatment
     causes the model-under-test to produce the expected change.
+
     :param base_test_case: A BaseTestCase object consisting of a treatment variable, outcome variable and effect
     :param expected_causal_effect: The expected causal effect (Positive, Negative, No Effect).
     :param effect_measure: A string which denotes the type of estimate to return.
@@ -70,10 +71,11 @@ def measure_adequacy(
     ) -> DataAdequacy:
         """
         Calculate the adequacy measurement, and populate the data_adequacy field.
+
         :param df: The original dataset to use.
         :param bootstrap_size: The number of bootstrap samples to use. (Defaults to 100)
         :param group_by: For IPCWEstimator - the "id" column to ensure that entire individuals are sampled rather than
-        random rows.
+                         random rows.
         """
         results = []
         outcomes = []
@@ -130,8 +132,7 @@ def execute_test(
         :param suppress_estimation_errors: Set to True to suppress estimation errors. (Defaults to False)
         :param bootstrap_size: The number of bootstrap samples to use. (Defaults to 100)
         :param group_by: For IPCWEstimator - the "id" column to ensure that entire individuals are sampled rather than
-        random rows.
-        :return causal_test_result: A CausalTestResult for the executed causal test case.
+                         random rows.
         """
         if not self.skip:
             try:
@@ -161,7 +162,8 @@ def estimate_effect(self, df: pd.DataFrame) -> CausalTestResult:
         Execute a causal test case and return the causal test result.
 
         :param df: The data to use.
-        :return causal_test_result: A CausalTestResult for the executed causal test case.
+
+        :returns: A CausalTestResult for the executed causal test case.
         """
         if self.query:
             df = df.query(self.query)

docs/source/modules/estimators.rst

@@ -31,7 +31,7 @@ LogisticRegressionEstimator
    :noindex:
 
 MultinomialRegressionEstimator
-~~~~~~~~~~~~~~~~~~~~~~~~~~~
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 **Recommended use:** For categorical outcomes (e.g. colurs: Red, Green, Blue).
 
@@ -42,19 +42,6 @@ MultinomialRegressionEstimator
    :show-inheritance:
    :noindex:
 
-CubicSplineRegressionEstimator
-~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
-
-**Recommended use:** For continuous outcomes with non-linear relationships or changes in behaviour.
-Useful when the relationship between treatment and outcome cannot be captured by a linear model.
-
-.. autoclass:: causal_testing.estimation.cubic_spline_estimator.CubicSplineRegressionEstimator
-   :members:
-   :undoc-members:
-   :show-inheritance:
-   :noindex:
-
-
 InstrumentalVariableEstimator
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~