def set_style(font_scale=None): # use default Latex font for math even with matplotlib 2.0 mpl.rcParams['mathtext.fontset'] = 'cm' if font_scale is not None: mpl.rcParams['font.size'] = 10 * font_scale