Feature/visualization #8

	# Install required Python dependencies, run tests, and lint with a single
	# version of Python. For more information see:
	# https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions

	name: Unit Tests

	on:
	pull_request:
	branches: [main]

	jobs:
	unittests:
	name: Unit Tests
	runs-on: ubuntu-latest
	steps:
	- name: Checkout code
	uses: actions/checkout@v2
	- name: Update packages
	run: sudo apt-get update
	- name: Set up Python 3.8
	uses: actions/setup-python@v2
	with:
	python-version: "3.8"

	- name: Install dependencies
	run: \|
	python3 -m pip install --upgrade pip
	pip3 install numpy
	pip3 install matplotlib
	pip3 install tqdm
	pip3 install torch>=1.7.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html --use-deprecated=legacy-resolver

	- name: Test locally with unittest
	run: python3 -m unittest

Provide feedback