Use iterkeys() to iterate dict keys

romanvm · Nov 20, 2017 · 3833a3e · 3833a3e
1 parent 84ee28e
commit 3833a3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -84,9 +84,9 @@ def get_categories():
         instead of returning lists.
 
     :return: The list of video categories
-    :rtype: list
+    :rtype: types.GeneratorType
     """
-    return VIDEOS.keys()
+    return VIDEOS.iterkeys()
 
 
 def get_videos(category):