tests

small fix for multistep ensemble training (#36) #178

	name: tests

	on: [push]

	jobs:
	build:

	runs-on: ubuntu-latest

	steps:
	- uses: actions/checkout@v4
	- name: Set up Python 3.12
	uses: actions/setup-python@v4
	with:
	python-version: '3.12'
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip setuptools wheel
	python -m pip install tqdm numpy numba parameterized xarray xskillscore timm jsbeautifier pynvml h5py wandb ruamel.yaml moviepy tensorly tensorly-torch more_itertools importlib-metadata
	python -m pip install torch==2.7.1 --extra-index-url https://download.pytorch.org/whl/cpu
	python -m pip install git+https://github.com/NVIDIA/[email protected]
	python -m pip install git+https://github.com/NVIDIA/[email protected]
	- name: Install package
	run: \|
	python -m pip install -e .
	- name: Test with pytest
	run: \|
	python -m pip install pytest pytest-cov parameterized
	python -m pytest ./tests

Provide feedback