Rework of train_model function #566

	name: 'nbdev-ci'
	on: [workflow_dispatch, pull_request, push]

	jobs:
	test:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v3
	- uses: actions/setup-python@v4
	with:
	python-version: '3.9'
	cache: "pip"
	cache-dependency-path: settings.ini
	- name: Test with nbdev
	shell: bash
	run: \|
	set -ux
	python -m pip install --upgrade pip
	pip install -U nbdev
	echo "Doing editable install..."
	test -f setup.py && pip install -e ".[all]"
	echo "Testing notebooks"
	nbdev_test

Provide feedback