diff --git a/src/probabilistic_model/__init__.py b/src/probabilistic_model/__init__.py index 180b280..e817444 100644 --- a/src/probabilistic_model/__init__.py +++ b/src/probabilistic_model/__init__.py @@ -1 +1 @@ -__version__ = "1.5.19" +__version__ = "1.5.20" diff --git a/src/probabilistic_model/probabilistic_model.py b/src/probabilistic_model/probabilistic_model.py index feb0d09..b92c7a7 100644 --- a/src/probabilistic_model/probabilistic_model.py +++ b/src/probabilistic_model/probabilistic_model.py @@ -228,20 +228,20 @@ class ProbabilisticModelWrapper: model: ProbabilisticModel """The model that is wrapped.""" - def _likelihood(self, event: Iterable) -> float: - return self.model._likelihood(event) + def likelihood(self, event: Iterable) -> float: + return self.model.likelihood(event) - def _probability(self, event: EncodedEvent) -> float: - return self.model._probability(event) + def probability(self, event: Event) -> float: + return self.model.probability(event) - def _mode(self) -> Tuple[Iterable[EncodedEvent], float]: - return self.model._mode() + def mode(self) -> Tuple[List[Event], float]: + return self.model.mode() def marginal(self, variables: Iterable[Variable]) -> Optional[Self]: return self.model.marginal(variables) - def _conditional(self, event: EncodedEvent) -> Tuple[Optional[Self], float]: - return self.model._conditional(event) + def conditional(self, event: Event) -> Tuple[Optional[Self], float]: + return self.model.conditional(event) def sample(self, amount: int) -> Iterable: return self.model.sample(amount)