Add argument in torch trainer whether to predict or not #42

	name: Version Branch CI/CD

	on:
	pull_request:
	branches: ["v*"]

	jobs:
	pytest:
	runs-on: ubuntu-latest
	container:
	image: python:3.11-slim
	env:
	NODE_VERSION: 20

	strategy:
	fail-fast: false
	matrix:
	python-version: ["3.10", "3.11", "3.12"]

	steps:
	- name: Install Node.js
	uses: actions/setup-node@v3
	with:
	node-version: ${{ env.NODE_VERSION }}

	- uses: actions/checkout@v3

	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v3
	with:
	python-version: ${{ matrix.python-version }}

	- name: Create virtual environment
	run: python -m venv venv

	- name: Activate virtual environment
	run: \|
	. venv/bin/activate

	- name: Install dependencies
	run: \|
	venv/bin/python -m pip install --upgrade pip
	venv/bin/python -m pip install pytest
	venv/bin/python -m pip install -r requirements.txt
	venv/bin/python -m pip install pytest-cov coverage

	- name: Test with pytest
	run: \|
	venv/bin/python -m pytest --cov=epochalyst --cov-branch --cov-fail-under=95 tests

	build:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v3

	- uses: actions/setup-python@v3
	with:
	python-version: 3.11

	- run: \|
	pip install build
	python -m build

	- uses: actions/upload-artifact@v3
	with:
	path: ./dist

Provide feedback