fix on_epoch_end #466

	name: CI

	on: [push]

	jobs:
	build:
	strategy:
	matrix:
	os: [ubuntu-22.04]
	pyv: [ '3.10', 3.11, 3.12]
	max-parallel: 5
	runs-on: ${{ matrix.os }}
	permissions:
	id-token: write
	steps:
	- uses: actions/checkout@v4
	with:
	fetch-depth: 0
	- name: Set up Python ${{ matrix.pyv }}
	run: \|
	conda install -y python=${{ matrix.pyv }}
	- name: Add conda to system path
	run: \|
	# $CONDA is an environment variable pointing to the root of the miniconda directory
	echo $CONDA/bin >> $GITHUB_PATH
	- name: Install dependencies
	run: \|
	conda env update --file environment.yml --name base --solver=classic
	conda install flake8
	# stop the build if there are Python syntax errors or undefined names
	flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
	# exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
	flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
	- name: Install with pip
	run: \|
	pip install -e .
	- name: Test with pytest
	run: \|
	conda install pytest
	pytest

Provide feedback