Fixed wrapper models.

tomsch420 · Jan 16, 2024 · 306713b · 306713b
1 parent 58edece
commit 306713b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git 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
@@ -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)