Refactoring Model Paralelisim over GPUs #903

	name: Python package

	on: [push,pull_request]
	jobs:
	build:

	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: ["3.10.13"]

	steps:
	- uses: actions/checkout@v3
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -e .["dev"]

	- name: Lint with ruff
	run: \|
	ruff check dicee/ --line-length=200
	- name: Test with pytest
	run: \|
	wget https://files.dice-research.org/datasets/dice-embeddings/KGs.zip --no-check-certificate && unzip KGs.zip
	python -m pytest -p no:warnings -x
	- name: Coverage report
	run: \|
	pip install coverage
	coverage run -m pytest -p no:warnings -x
	coverage report -m

Provide feedback