Minor Fix

MitchellShibilski-Unkel · May 8, 2024 · 0d928c9 · 0d928c9
1 parent c638371
commit 0d928c9
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/PyGrammar/PyGrammar.py b/PyGrammar/PyGrammar.py
@@ -1,4 +1,6 @@
 import nltk 
+import re
+import random
 import language_tool_python as lang
 
 
@@ -16,6 +18,7 @@ def getPartsOfSpeech(self):
         return POS
 
     def correct(self):
+        self.lang.enable_spellchecking()
         getPOS = self.getPartsOfSpeech()
 
         sentences: list = self.text.split(".")
@@ -34,4 +37,6 @@ def correct(self):
 
             correctedText = str(re.sub(r'\b(\w+)\s+\1\b', r'\1', correctedText))
 
-        return self.lang.correct(correctedText.replace(" .", ".").replace("\'", ""))
+        correctedText.replace(" .", ".").replace("\'", "")
+
+        return self.lang.correct(correctedText)
diff --git a/PyGrammar/__pycache__/PyGrammar.cpython-310.pyc b/PyGrammar/__pycache__/PyGrammar.cpython-310.pyc
diff --git a/PyGrammar/__pycache__/__init__.cpython-310.pyc b/PyGrammar/__pycache__/__init__.cpython-310.pyc