GitHub Actions workflow, and modernize hackable_diffusion build: replace requirements.txt with pyproject.toml

PiperOrigin-RevId: 890419375

Author: Qwlouse

.github/workflows/pytest_and_autopublish.yml

@@ -0,0 +1,34 @@
+name: Unittests & Auto-publish
+
+# Allow to trigger the workflow manually (e.g. when deps changes)
+on: [push, workflow_dispatch]
+
+jobs:
+  pytest-job:
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ['3.13']
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+
+    concurrency:
+      group: ${{ github.workflow }}-${{ github.ref }}-${{ matrix.python-version }}
+      cancel-in-progress: true
+
+    steps:
+    - uses: actions/checkout@v3
+
+    # Install deps
+    - uses: actions/setup-python@v4
+      with:
+        python-version: ${{ matrix.python-version }}
+    - run: pip --version
+    - run: pip install --no-cache-dir -e .[dev]
+    - run: pip freeze
+
+    # Run tests
+    - name: Run core tests
+      run: pytest -vv -n auto
+
+  # TODO(klausg): Add publish job to push to PyPI later

README.md

@@ -25,10 +25,16 @@ The `notebooks/` directory contains several tutorials to get you started:
 ## Installation
 
 To install the necessary dependencies, you can use pip with the provided
-`requirements.txt` file:
+`pyproject.toml` file:
 
 ```bash
-pip install -r requirements.txt
+pip install -e .
+```
+
+To install development dependencies (for running tests), use:
+
+```bash
+pip install -e .[dev]
 ```
 
 This will install libraries such as JAX, Flax, and other utilities required to

pyproject.toml

@@ -0,0 +1,87 @@
+[project]
+name = "hackable_diffusion"
+description = "Hackable Diffusion is a library for custom diffusion models."
+readme = "README.md"
+requires-python = ">=3.13"
+license = {file = "LICENSE"}
+authors = [{name = "hackable_diffusion authors"}]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "License :: OSI Approved :: Apache Software License",
+    "Intended Audience :: Science/Research",
+]
+keywords = []
+
+# pip dependencies of the project
+dependencies = [
+    "absl-py",
+    "aiofiles",
+    "chex",
+    "einops",
+    "etils",
+    "flax",
+    "fsspec",
+    "humanize",
+    "immutabledict",
+    "importlib_resources",
+    "jax",
+    "jaxtyping",
+    "markdown-it-py",
+    "mdurl",
+    "ml_dtypes",
+    "mock",
+    "msgpack",
+    "nest-asyncio",
+    "numpy",
+    "opt_einsum",
+    "optax",
+    "orbax-checkpoint",
+    "protobuf",
+    "psutil",
+    "pygments",
+    "pyyaml",
+    "rich",
+    "scipy",
+    "simplejson",
+    "tensorstore",
+    "toolz",
+    "treescope",
+    "typeguard",
+    "typing_extensions",
+    "wadler_lindig",
+    "zipp",
+    "kauldron>=1.4.1",
+]
+
+# This is set automatically by flit using `hackable_diffusion.__version__`
+dynamic = ["version"]
+
+[project.urls]
+homepage = "https://github.com/google-research/hackable_diffusion"
+repository = "https://github.com/google-research/hackable_diffusion"
+
+[project.optional-dependencies]
+# Development deps (unittest, linting, formating,...)
+dev = [
+    "pytest",
+    "pytest-xdist",
+    "hypothesis==6.80.0",
+    "pylint>=2.6.0",
+    "pyink",
+]
+
+[build-system]
+requires = ["flit_core >=3.8,<4"]
+build-backend = "flit_core.buildapi"
+
+[tool.pyink]
+line-length = 80
+preview = true
+pyink-indentation = 2
+pyink-use-majority-quotes = true
+
+[tool.flit.sdist]
+exclude = [
+  "**/*_test.py",
+]