Feat/support to set device for batch dataset (#147) #60

	name: Test Python

	on: [push, pull_request]

	jobs:
	test:

	runs-on: ubuntu-latest

	steps:
	- uses: actions/checkout@v4
	- name: Set up Python
	uses: actions/setup-python@v3
	with:
	python-version: '3.x'
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install .
	pip install pytest
	- name: Test with pytest
	run: \|
	pytest

Provide feedback