oops

untitled/namegen.py

@@ -22,11 +22,11 @@ class NameGenerator:
                 model[word[pos] + word[pos + 1]].append(word[pos + 2])
         return model
 
-    def make_name(self, model):
+    def make_name(self):
         result = "<<"
         while True:
             window = result[-2:]
-            nxt = random.choice(model[window])
+            nxt = random.choice(self.model[window])
             if nxt == ">":
                 break
             result += nxt
@@ -34,6 +34,6 @@ class NameGenerator:
 
     def generate_name(self):
         while True:
-            name = self.make_name(self.model)
+            name = self.make_name()
             if self.validate and self.validate(name) is None:
                 return name