.

torch_struct/cky.py

@@ -309,7 +309,7 @@ def _intermediary(spans):
                         c_final = C_p
                         break
                 if b_final is not None: break
-            assert k_final is not None
+            assert k_final is not None, b, i, j
             splits[(b, i, j)] = k_final, b_final, c_final
         return splits