MLPF datasets v2.0.0: track pythia-level genjets, genmet in datasets; add per-particle ispu flag #1972

	name: CI

	on:
	push:
	branches: [main]
	pull_request:
	branches: [main]

	workflow_dispatch:

	jobs:
	deps:
	runs-on: ubuntu-22.04
	steps:
	- uses: actions/checkout@v3
	- uses: actions/setup-python@v4
	with:
	python-version: "3.10.12"
	cache: "pip"
	- run: pip install -r requirements.txt
	deps-pyg:
	runs-on: ubuntu-20.04
	needs: [deps]
	steps:
	- uses: actions/checkout@v3
	- uses: actions/setup-python@v4
	with:
	python-version: "3.10.12"
	cache: "pip"
	- run: pip install -r requirements.txt
	- run: pip3 install torch==2.2.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

	pyg-unittests:
	runs-on: ubuntu-22.04
	needs: [deps-pyg]
	steps:
	- uses: actions/checkout@v3
	- uses: actions/setup-python@v4
	with:
	python-version: "3.10.12"
	cache: "pip"
	- run: pip install -r requirements.txt
	- run: pip3 install torch==2.2.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
	- run: PYTHONPATH=. python3 -m unittest tests/test_torch_and_tf.py

	pyg-pipeline:
	runs-on: ubuntu-22.04
	needs: [pyg-unittests]
	steps:
	- uses: actions/checkout@v3
	- uses: actions/setup-python@v4
	with:
	python-version: "3.10.12"
	cache: "pip"
	- run: pip install -r requirements.txt
	- run: pip3 install torch==2.2.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
	- run: ./scripts/local_test_pyg.sh

Provide feedback