A few small changes for being compatible with the latest prompt_toolk…

…it 2.0
j-bennet · Jan 31, 2018 · f6baac5 · f6baac5
1 parent ff53ad2
commit f6baac5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wharfee/style.py b/wharfee/style.py
@@ -2,7 +2,7 @@
 from pygments.token import Token
 from pygments.util import ClassNotFound
 from prompt_toolkit.styles import merge_styles
-from prompt_toolkit.styles.pygments import style_from_pygments, style_from_pygments_dict
+from prompt_toolkit.styles.pygments import style_from_pygments_cls, style_from_pygments_dict
 import pygments.styles
 
 
@@ -13,7 +13,7 @@ def style_factory(name):
         pygments_style = pygments.styles.get_style_by_name('native')
 
     return merge_styles([
-        style_from_pygments(pygments_style),
+        style_from_pygments_cls(pygments_style),
         style_from_pygments_dict({
             Token.Menu.Completions.Completion.Current: 'bg:#00aaaa #000000',
             Token.Menu.Completions.Completion: 'bg:#008888 #ffffff',