Fix: adding import nececeary for using attention layers as bottlenecks #310

	name: CI
	on: [push, pull_request]
	jobs:
	build:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v1
	- uses: actions/setup-python@v1
	with:
	python-version: '3.6'
	architecture: 'x64'
	- name: Install the library
	run: \|
	pip install jupyter
	pip install -e .
	pip install nbdev
	- name: Read all notebooks
	run: \|
	nbdev_read_nbs
	- name: Check if all notebooks are cleaned
	run: \|
	echo "Check we are starting with clean git checkout"
	if [ -n "$(git status -uno -s)" ]; then echo "git status is not clean"; false; fi
	echo "Trying to strip out notebooks"
	nbdev_clean_nbs
	echo "Check that strip out was unnecessary"
	git status -s # display the status to see which nbs need cleaning up
	if [ -n "$(git status -uno -s)" ]; then echo -e "!!! Detected unstripped out notebooks\n!!!Remember to run nbdev_install_git_hooks"; false; fi
	- name: Check if there is no diff library/notebooks
	run: \|
	if [ -n "$(nbdev_diff_nbs)" ]; then echo -e "!!! Detected difference between the notebooks and the library"; false; fi

Provide feedback