fixed arg parsing for stemming

uwsampa · May 26, 2015 · 56d1919 · 56d1919
1 parent 5cf7b04
commit 56d1919
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mapred_tfidf.py b/mapred_tfidf.py
@@ -76,8 +76,8 @@ def get_output_dir(sub_dir=''):
                         help=stop_words_help, dest='stop_words')
 
     stemmer_help = 'if true, use nltk PorterStemmer to stem ngrams'
-    parser.add_argument('--stem', type=bool, action='store_true', dest='stem')
-    parser.add_argument('--no-stem', type=bool, action='store_false', dest='stem')
+    parser.add_argument('--stem', action='store_true', dest='stem')
+    parser.add_argument('--no-stem', action='store_false', dest='stem')
     parser.set_defaults(stem=True)
 
     args = parser.parse_args()