fix restarting lammps simulations #1726

	name: CI

	on:
	push:
	branches:
	- main
	- trunk-merge/**
	paths-ignore:
	- "README.md"
	pull_request:
	branches:
	- main
	paths-ignore:
	- "README.md"
	workflow_call:

	jobs:
	test:
	runs-on: ${{ matrix.os }}
	strategy:
	fail-fast: false
	matrix:
	os: [ubuntu-latest, macos-latest]
	python-version: [3.9, 3.13]
	steps:
	- uses: actions/checkout@v4
	- name: Set up Python ${{ matrix.python-version }} on ${{ matrix.os }}
	uses: actions/setup-python@v5
	with:
	python-version: ${{ matrix.python-version }}

	- name: Install python dependencies
	env:
	JAX_MD_PKG: jax-md-v0.2.${{ matrix.python-version == '3.9' && 24 \|\| 25 }}.zip
	run: \|
	python -m pip install --upgrade pip
	pip install https://github.com/jax-md/jax-md/archive/refs/tags/${{ env.JAX_MD_PKG }}
	pip install ase dill pytest matplotlib

	- name: Install pysages
	run: pip install .

	- name: Run tests
	run: \|
	pytest
	mkdir /tmp/test_plots
	mv *.pdf /tmp/test_plots

	- name: Upload artifact
	uses: actions/upload-artifact@v4
	with:
	name: test_plots-${{ matrix.os }}-py${{ matrix.python-version }}
	path: /tmp/test_plots

	doc-test:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v4
	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: 3.9
	- name: Install python dependecies
	run: \|
	python -m pip install --upgrade pip
	pip install https://github.com/jax-md/jax-md/archive/refs/tags/jax-md-v0.2.24.zip
	pip install dill pytest pylint flake8
	pip install -r docs/requirements.txt
	- name: Install pysages
	run: pip install .
	- name: Build documentation
	run: cd docs && sphinx-build -b spelling -W --keep-going source build

Provide feedback