fix bug (#1311) #1536

	name: "Run CPU Tests"

	on: "push"

	jobs:
	run-tests:
	#runs-on: ubuntu-latest
	runs-on: ubuntu-22.04
	steps:
	- uses: actions/checkout@v3

	- name: Install Python
	uses: actions/setup-python@v4
	with:
	python-version: "3.8"
	cache: "pip"
	cache-dependency-path: "*/requirements.txt"

	- name: Upgrade Pip
	run: python -m pip install --upgrade pip

	- name: Install Dependencies
	run: \|
	sudo apt-get install libopenmpi-dev -y
	pip install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cpu
	pip install -r requirements/requirements.txt
	pip install -r requirements/requirements-dev.txt
	pip install -r requirements/requirements-wandb.txt

	- name: Prepare Data
	run: python prepare_data.py

	- name: Run CPU Tests
	run: PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python pytest tests -m cpu

Provide feedback