chore: modernized build

Author: sehoffmann

.github/actions/setup_environment/action.yml

@@ -7,13 +7,13 @@ inputs:
 runs:
   using: composite
   steps:
-    - name: Setup Python
-      uses: actions/setup-python@v5
+    - name: Install uv
+      uses: astral-sh/setup-uv@v6
       with:
+        version: "0.8.9"
         python-version: ${{ inputs.python-version }}
+        enable-cache: true
 
-    - name: Install CI Dependencies
+    - name: Install the project
       shell: bash
-      run: |
-        pip install -r ci_requirements.txt
-        echo "/home/runner/.local/bin" >> $GITHUB_PATH
+      run: uv sync --locked --all-extras --dev

.github/workflows/build_and_upload_wheel.yml

@@ -20,7 +20,7 @@ jobs:
         uses: ./.github/actions/setup_environment
 
       - name: Lint
-        run: pre-commit run --all-files
+        run: uv run pre-commit run --all-files
 
       - name: Required modifications
         if: ${{ failure() }}

.github/workflows/release_public.yml

@@ -28,19 +28,11 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Install Dependencies
-        run: |
-          pip install -r requirements.txt
-
-      - name: Install Project
-        run: |
-          pip install .
-
       - name: Run tests & coverage
         run: |
-          coverage run -m pytest --no-header -v test
-          coverage report -m -i
-          coverage html -i
+          uv run coverage run -m pytest --no-header -v test
+          uv run coverage report -m -i
+          uv run coverage html -i
 
       - name: Archive coverage results
         if: startsWith(matrix.os, 'ubuntu')

.github/workflows/release_test.yml

@@ -1,26 +1,26 @@
 
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.5.0
+    rev: v5.0.0
     hooks:
     -   id: trailing-whitespace
     -   id: end-of-file-fixer
 
   - repo: https://github.com/asottile/pyupgrade
-    rev: v3.15.2
+    rev: v3.20.0
     hooks:
       - id: pyupgrade
         args: [--py37-plus]
 
   - repo: https://github.com/omnilib/ufmt
-    rev: v2.5.1
+    rev: v2.8.0
     hooks:
       - id: ufmt
         additional_dependencies:
-          - black == 23.1.0
+          - black == 25.1.0
           - usort == 1.1.0b2
 
   - repo: https://github.com/pycqa/flake8
-    rev: 7.0.0
+    rev: 7.2.0
     hooks:
       - id: flake8

.github/workflows/run_linting.yml

@@ -1,42 +1,50 @@
 [build-system]
-requires = ["setuptools", "packaging", "wheel", "build", "pre-commit", "pytest"]
-build-backend = "setuptools.build_meta"
+requires = ["hatchling"]
+build-backend = "hatchling.build"
 
 [project]
 name = "dmlcloud"
+description = "Painless distributed training with torch"
+version = "0.4"
 authors = [
-    {name = "Sebastian Hoffmann"}
+    { name = "Sebastian Hoffmann", email = "[email protected]"},
 ]
-description = "Distributed torch training using horovod and slurm"
+maintainers = [
+    { name = "Sebastian Hoffmann", email = "[email protected]"},
+]
+readme = "README.md"
+license-files = ["LICENSE"]
+
 requires-python = ">=3.10"
-license = {file = "LICENSE"}
-keywords = ["pytorch", "torch.distributed", "slurm", "distributed training", "deep learning"]
-classifiers = [
-    "Development Status :: 3 - Alpha",
-    "License :: OSI Approved :: BSD License",
-    "Operating System :: MacOS",
-    "Operating System :: POSIX :: Linux",
-    "Programming Language :: Python :: 3.10",
-    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+dependencies = [
+    "torch~=2.5",
+    "torchmetrics~=1.6",
+    "progress_table>=2.2.0",
+    "numpy",
+    "xarray",
+    "nvidia-ml-py",
 ]
-dynamic = ["version", "readme", "dependencies"]
 
 [project.urls]
 Repository = "https://github.com/sehoffmann/dmlcloud"
 
 [project.scripts]
 dmlrun = "dmlcloud.run:main"
 
-[tool.setuptools.packages.find]
-include = ["dmlcloud*"]
-namespaces = false
-
-[tool.setuptools.dynamic]
-version = {attr = "dmlcloud.__version__"}
-readme = {file = ["README.md"], content-type = "text/markdown"}
-dependencies = {file = ["requirements.txt"]}
+[dependency-groups]
+dev = [
+    "pre-commit",
+    "pytest",
+    "coverage",
+    "sphinx",
+    "sphinx-rtd-theme",
+    "sphinx-autodoc-typehints",
+]
 
 [tool.black]
 skip-string-normalization = true
 line-length = 120
-target-version = ["py310"]
+target-version = ["py310", "py311", "py312", "py313"]
+
+[tool.uv]
+python-preference = "managed"