From 21d6efd2fd29f20e3c8f6d15fe8a63c00b6e72d3 Mon Sep 17 00:00:00 2001 From: jessegrabowski Date: Thu, 12 Jun 2025 16:15:13 -0700 Subject: [PATCH] Add `BaseEstimator` mixin to `LinearModel` --- pymc_extras/linearmodel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pymc_extras/linearmodel.py b/pymc_extras/linearmodel.py index b1f78b263..639cd1749 100644 --- a/pymc_extras/linearmodel.py +++ b/pymc_extras/linearmodel.py @@ -2,10 +2,12 @@ import pandas as pd import pymc as pm +from sklearn.base import BaseEstimator + from pymc_extras.model_builder import ModelBuilder -class LinearModel(ModelBuilder): +class LinearModel(ModelBuilder, BaseEstimator): def __init__( self, model_config: dict | None = None, sampler_config: dict | None = None, nsamples=100 ):