1 files changed, 7 insertions, 2 deletions
diff --git a/rst_parser/rst_test.cc b/rst_parser/rst_test.cc
index e8fe706e..8995515f 100644
--- a/rst_parser/rst_test.cc
+++ b/rst_parser/rst_test.cc
@@ -26,8 +26,13 @@ int main(int argc, char** argv) {
   af(0,2).edge_prob.logeq(10);
   af(0,1).edge_prob.logeq(9);
   af(0,3).edge_prob.logeq(9);
-  SpanningTree tree;
-  af.MaximumSpanningTree(&tree);
+  EdgeSubset tree;
+//  af.MaximumEdgeSubset(&tree);
+  double lz;
+  af.EdgeMarginals(&lz);
+  cerr << "Z = " << lz << endl;
+  af.PickBestParentForEachWord(&tree);
+  cerr << tree << endl;
   return 0;
 }