add _ after fit variables

wd60622 · Jan 14, 2024 · 34045ff · 34045ff
1 parent a7ab311
commit 34045ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/latent_calendar/model/latent_calendar.py b/latent_calendar/model/latent_calendar.py
@@ -177,11 +177,11 @@ def fit(self, X, y=None) -> "ConjugateModel":
         if self.a is None:
             self.a = hourly_prior(X)
 
-        self.prior = Dirichlet(alpha=self.a)
+        self.prior_ = Dirichlet(alpha=self.a)
         return self
 
     def transform(self, X, y=None) -> np.ndarray:
-        return multinomial_dirichlet(X, self.prior).dist.mean()
+        return multinomial_dirichlet(X, self.prior_).dist.mean()
 
     def predict(self, X, y=None) -> np.ndarray:
         return self.transform(X, y=y)