.

harvardnlp · Oct 16, 2019 · d198da0 · d198da0
1 parent 95d0f53
commit d198da0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/torch_struct/__init__.py b/torch_struct/__init__.py
@@ -16,6 +16,7 @@
 from .semirings import (
     LogSemiring,
     StdSemiring,
+    KMaxSemiring,
     SampledSemiring,
     MaxSemiring,
     EntropySemiring,
@@ -36,6 +37,7 @@
     StdSemiring,
     SampledSemiring,
     MaxSemiring,
+    KMaxSemiring,
     EntropySemiring,
     MultiSampledSemiring,
     SelfCritical,

diff --git a/torch_struct/distributions.py b/torch_struct/distributions.py
@@ -6,7 +6,7 @@
 from .semimarkov import SemiMarkov
 from .deptree import DepTree, deptree_nonproj, deptree_part
 from .cky_crf import CKY_CRF
-from .semirings import LogSemiring, MaxSemiring, EntropySemiring, MultiSampledSemiring
+from .semirings import LogSemiring, MaxSemiring, EntropySemiring, MultiSampledSemiring, KMaxSemiring
 
 
 class StructDistribution(Distribution):