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

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

(limited to 'gen.cpp')

diff --git a/gen.cpp b/gen.cpp
index eba0277..5e2d9db 100644
--- a/gen.cpp
+++ b/gen.cpp
@@ -69,14 +69,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::cout << "Generating..." << std::endl;
-- 
cgit 1.4.1