removed reverse bool

GoogleCloudPlatform · Jul 20, 2016 · d465566 · jerjou · Jul 20, 2016 · d465566
1 parent c9ac6d3
commit d465566
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/language/movie_nl/main.py b/language/movie_nl/main.py
@@ -282,12 +282,9 @@ def rank_entities(reader, sentiment=None, topn=None, reverse_bool=False):
         else:
             items.append(entity_item)
 
-    items.sort(reverse=True)
+    items.sort(reverse=reverse_bool)
     items = [json.dumps(item[1]) for item in items]
 
-    if reverse_bool:
-        items.reverse()
-
     if topn:
         print('\n'.join(items[:topn]))
     else:
@@ -340,7 +337,9 @@ def analyze(input_dir, sentiment_writer, entity_writer, sample, log_file):
     rank_parser.add_argument(
         '--sentiment', help='filter sentiment as "neg" or "pos"')
     rank_parser.add_argument(
-        '--reverse', help='reverse the order of the items')
+        '--reverse', help='reverse the order of the items', type=bool,
+        default=False
+        )
     rank_parser.add_argument(
         '--sample', help='number of top items to process', type=int)