.github: add nightly wheel builds

Author: d4l3k

.github/workflows/nightly.yaml

@@ -0,0 +1,47 @@
+name: Nightly Build
+
+on:
+  pull_request:
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.10"
+          architecture: x64
+      - name: Checkout
+        uses: actions/checkout@v4
+      - name: Install Dependencies
+        run: |
+          set -eux
+
+          pip install -U twine toml
+      - name: Build Docker
+        run: |
+          set -eux
+
+          docker build --progress=plain -t torchft-maturin .
+
+      - name: Set Nightly Version
+        run: |
+          set -eux
+
+          python scripts/patch_nightly_version.py
+
+      - name: Build Wheels
+        run: |
+          set -eux
+
+          docker run --rm -v $(pwd):/io -t torchft-maturin build --release --out dist --interpreter 3.10
+
+      - name: Twine Check
+        run: twine check --strict dist/*
+
+      - name: Upload to Pypi
+        run: twine upload dist/*
+        env:
+          TWINE_USERNAME: __token__
+          TWINE_PASSWORD: ${{ secrets.pypi_token }}

.gitignore

@@ -26,6 +26,7 @@ __pycache__
 *.so
 **/*.stderr
 .pyre
+dist/
 
 # Torch
 cifar/

build.rs

@@ -7,6 +7,6 @@
 fn main() -> Result<(), Box<dyn std::error::Error>> {
     tonic_build::configure()
         .protoc_arg("--experimental_allow_proto3_optional")
-        .compile_protos(&["proto/torchft.proto"], &["proto"])?;
+        .compile(&["proto/torchft.proto"], &["proto"])?;
     Ok(())
 }

scripts/patch_nightly_version.py

@@ -0,0 +1,26 @@
+from datetime import date
+
+import toml
+
+
+def get_nightly_version():
+    today = date.today()
+    return f"{today.year}.{today.month}.{today.day}"
+
+
+CARGO_FILE = "Cargo.toml"
+PYPROJECT_FILE = "pyproject.toml"
+
+
+cargo = toml.load(CARGO_FILE)
+cargo["package"]["version"] = get_nightly_version()
+
+print(cargo)
+
+with open(CARGO_FILE, "w") as f:
+    toml.dump(cargo, f)
+
+pyproject = toml.load(PYPROJECT_FILE)
+pyproject["project"]["name"] = "torchft-nightly"
+
+print(pyproject)