From 4d217ac6122120d9e86248432594864e114e3a46 Mon Sep 17 00:00:00 2001
From: Kelly Rauchenberger <fefferburbia@gmail.com>
Date: Thu, 26 May 2016 21:46:41 -0400
Subject: Fixed ebooks and gen ignoring the form distribution

---
 ebooks.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'ebooks.cpp')

diff --git a/ebooks.cpp b/ebooks.cpp
index 8b567ed..52247da 100644
--- a/ebooks.cpp
+++ b/ebooks.cpp
@@ -64,14 +64,14 @@ int main(int argc, char** args)
   rawr kgramstats;
   kgramstats.addCorpus(corpus);
   kgramstats.compile(4);
-  kgramstats.setTransformCallback([&] (std::string canonical, std::string) {
-    size_t pos = canonical.find("$name$");
+  kgramstats.setTransformCallback([&] (std::string, std::string form) {
+    size_t pos = form.find("$name$");
     if (pos != std::string::npos)
     {
-      canonical.replace(pos, 6, fv_names[rand() % fv_names.size()]);
+      form.replace(pos, 6, fv_names[rand() % fv_names.size()]);
     }
     
-    return canonical;
+    return form;
   });
   
   std::mutex stats_mutex;
-- 
cgit 1.4.1