Fix log partition function name (#96)

Co-authored-by: John Reid <[email protected]>
harvardnlp · Feb 12, 2021 · c875d93 · c875d93
1 parent e3b04e3
commit c875d93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/alignment.py b/torch_struct/alignment.py
@@ -45,7 +45,7 @@ def _check_potentials(self, edge, lengths=None):
         assert max(lengths) == N, "One length must be at least N"
         return edge, batch, N, M, lengths
 
-    def logparition(self, log_potentials, lengths=None, force_grad=False, cache=True):
+    def logpartition(self, log_potentials, lengths=None, force_grad=False, cache=True):
         return self._dp_scan(log_potentials, lengths, force_grad)
 
     def _dp_scan(self, log_potentials, lengths=None, force_grad=False):