Fix problematic sample in Schelling Point #1752

	name: Run unit tests

	on:
	pull_request:
	branches:
	- main
	push:
	branches:
	- main

	jobs:
	check_files:
	runs-on: ubuntu-latest

	steps:
	- name: Checkout repository
	uses: actions/checkout@v2
	with:
	fetch-depth: 0
	lfs: true

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

	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install pyyaml
	pip install pytest
	pip install -e .[torch]

	- name: Run unit tests
	env:
	OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
	run: \|
	pytest

Provide feedback