Make the spe *_mem methods part of the API.

probsys · Jan 5, 2022 · fa8670d · fa8670d
1 parent cc557c1
commit fa8670d
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/spe.py b/src/spe.py
@@ -86,6 +86,14 @@ def logpdf(self, assignment, memo=None):
         raise NotImplementedError()
     def constrain(self, assignment, memo=None):
         raise NotImplementedError()
+    def logprob_mem(self, event_factor, memo):
+        raise NotImplementedError()
+    def condition_mem(self, event_factor, memo):
+        raise NotImplementedError()
+    def logpdf_mem(self, assignment, memo):
+        raise NotImplementedError()
+    def constrain_mem(self, assignment, memo):
+        raise NotImplementedError()
     def mutual_information(self, A, B, memo=None):
         if memo is None:
             memo = Memo()
@@ -190,14 +198,6 @@ def constrain(self, assignment, memo=None):
         if memo is None:
             memo = Memo()
         return self.constrain_mem(assignment, memo)
-    def logprob_mem(self, event_factor, memo):
-        raise NotImplementedError()
-    def condition_mem(self, event_factor, memo):
-        raise NotImplementedError()
-    def logpdf_mem(self, assignment, memo):
-        raise NotImplementedError()
-    def constrain_mem(self, assignment, memo):
-        raise NotImplementedError()
 
 # ==============================================================================
 # Sum SPE.