Checking input is string or list of strings

sgrau · Sep 11, 2012 · cc4944b · cc4944b
1 parent 76d8791
commit cc4944b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/keywords/keywords.py b/keywords/keywords.py
@@ -17,7 +17,11 @@ def keywords_and_ngrams(input, nKeywords=100, thresholdLL=19, nBigrams=25, thres
     fdistText = defaultdict(int)
     sumText = 0
     listWords = []
-
+    listSentences = []
+    if isinstance(input,str):
+        listSentences = [input]
+    else:
+        listSentences = input
     for line in input:
         for w in line.split():
             w = w.lower()