from bokeh.plotting import figure, output_file, show
from bokeh.io import output_notebook
from bokeh.models import ColumnDataSource

# Set Bokeh output to notebook
output_notebook()