Changing github action to install torch for CPU #95

	name: Continuous Integration

	on:
	push:
	# edit for testing
	branches: [ 'main', 'develop', '28-notebooks-error']
	pull_request:
	branches: [ 'main', 'develop' ]

	jobs:

	# --------------------------------
	# Lint the Codebase
	# --------------------------------

	lint:
	name: Lint
	runs-on: ubuntu-latest
	steps:
	- name: Checkout
	uses: actions/checkout@v3

	- name: Format
	uses: psf/black@stable


	# --------------------------------
	# Build the Package
	# --------------------------------

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

	- name: Install dependency manager
	uses: snok/install-poetry@v1

	- name: Install dependencies
	run: make install-dependencies

	- name: Build package
	run: make build

	# --------------------------------
	# Test if notebooks build
	# --------------------------------

	run_notebooks:
	runs-on: ubuntu-latest

	steps:
	- name: Checkout repository
	uses: actions/checkout@v2

	- name: Set up Python
	uses: actions/setup-python@v2
	with:
	python-version: 3.8

	- name: Install Poetry
	run: \|
	pip install poetry
	poetry install

	- name: Install PyTorch (CPU only)
	run: poetry add torch==1.10.0+cpu torchvision==0.11.1+cpu torchaudio==0.10.0+cpu -D

	- name: Run Notebooks
	run: poetry run jupyter nbconvert --execute notebooks/*/.ipynb

Provide feedback