Follow-up for #1854. Fix migration guide update. Bump tf-keras to 2.1… #6626

	name: Continuous integration

	on:
	push:
	branches: [master]
	pull_request:

	jobs:
	black:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v2
	- uses: actions/setup-python@v1
	with:
	python-version: 3.9
	- name: Ensure files are formatted with black
	run: \|
	pip install --upgrade pip
	pip install black --upgrade
	black --check --diff ./examples
	docker-image:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v2
	- name: Ensure the docker image works and can start.
	run: \|
	make container-test

Provide feedback