pylint

Author: jmafoster1

causal_testing/estimation/cubic_spline_estimator.py

@@ -1,23 +1,12 @@
-"""This module contains the CubicSplineRegressionEstimator class, for estimating continuous outcomes with changes in behaviour"""
+"""This module contains the CubicSplineRegressionEstimator class, for estimating
+continuous outcomes with changes in behaviour"""
 
 import logging
-from abc import ABC, abstractmethod
 from typing import Any
-from math import ceil
 
-import numpy as np
 import pandas as pd
-import statsmodels.api as sm
-import statsmodels.formula.api as smf
-from patsy import dmatrix  # pylint: disable = no-name-in-module
-from patsy import ModelDesc
-from statsmodels.regression.linear_model import RegressionResultsWrapper
-from statsmodels.tools.sm_exceptions import PerfectSeparationError
-from lifelines import CoxPHFitter
 
 from causal_testing.specification.variable import Variable
-from causal_testing.specification.capabilities import TreatmentSequence, Capability
-from causal_testing.estimation.estimator import Estimator
 from causal_testing.estimation.linear_regression_estimator import LinearRegressionEstimator
 
 logger = logging.getLogger(__name__)

causal_testing/estimation/estimator.py

@@ -3,20 +3,8 @@
 import logging
 from abc import ABC, abstractmethod
 from typing import Any
-from math import ceil
 
-import numpy as np
 import pandas as pd
-import statsmodels.api as sm
-import statsmodels.formula.api as smf
-from patsy import dmatrix  # pylint: disable = no-name-in-module
-from patsy import ModelDesc
-from statsmodels.regression.linear_model import RegressionResultsWrapper
-from statsmodels.tools.sm_exceptions import PerfectSeparationError
-from lifelines import CoxPHFitter
-
-from causal_testing.specification.variable import Variable
-from causal_testing.specification.capabilities import TreatmentSequence, Capability
 
 logger = logging.getLogger(__name__)