feat: remove extra arch + refactor unit tests workflow #4

	name: Unittests

	on:
	pull_request:
	types: [opened, reopened, synchronize, ready_for_review]
	push:
	branches:
	- "master"

	jobs:
	unittests:
	strategy:
	env:
Check failure on line 13 in .github/workflows/tests.yaml View workflow run for this annotation GitHub Actions / Unittests Invalid workflow file `The workflow is not valid. .github/workflows/tests.yaml (Line: 13, Col: 7): Unexpected value 'env'`
	os: [ubuntu-latest]
	python-version: [3.10]
	torch-version: [2.0.1, 2.1.1, 2.2.1, 2.3.1, 2.4.1]
	runs-on: ${{ matrix.os }}
	steps:
	- uses: actions/checkout@v2
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v2
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install numpy scikit-learn flake8 setuptools numba
	- name: Install Torch ${{ matrix.torch-version }}
	run: pip install torch==${{ matrix.torch-version }}+cpu -f https://download.pytorch.org/whl/torch_stable.html
	- name: Build package
	run: \|
	python setup.py build_ext --inplace
	- name: Lint with flake8
	run: \|
	# 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: Test with unittest
	run: \|
	python -m unittest -v

Provide feedback