Fix dataset mixing logic #929

	name: Tests

	on:
	push:
	branches:
	- main
	- v*-release
	pull_request:
	branches:
	- main

	jobs:

	unit-tests:
	name: Run unit tests
	runs-on: ubuntu-latest
	steps:
	- name: Checkout code
	uses: actions/checkout@v2
	- name: Setup Python environment
	uses: actions/setup-python@v2
	with:
	python-version: 3.10.10
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	python -m pip install pytest torch datasets transformers==4.40.2 accelerate rich
	- name: Run unit tests
	run: pytest open_instruct/test_utils.py

Provide feedback