diff --git a/torch_struct/__init__.py b/torch_struct/__init__.py
index 42bd7479..02c4122a 100644
--- a/torch_struct/__init__.py
+++ b/torch_struct/__init__.py
@@ -1,6 +1,18 @@
 from .cky import cky_inside, cky
-from .deptree import deptree_inside, deptree, deptree_nonproj, deptree_toseq, deptree_fromseq
-from .linearchain import linearchain, linearchain_forward, hmm, linearchain_fromseq,   linearchain_toseq
+from .deptree import (
+    deptree_inside,
+    deptree,
+    deptree_nonproj,
+    deptree_toseq,
+    deptree_fromseq,
+)
+from .linearchain import (
+    linearchain,
+    linearchain_forward,
+    hmm,
+    linearchain_fromseq,
+    linearchain_toseq,
+)
 
 from .semimarkov import semimarkov, semimarkov_forward
 from .semirings import LogSemiring, StdSemiring, SampledSemiring, MaxSemiring