.github: add nightly wheel builds

.github/workflows/nightly.yaml

@@ -0,0 +1,62 @@
+name: Nightly Push
+
+on:
+  # run every day at 11:15am
+  schedule:
+    - cron:  '15 11 * * *'
+
+jobs:
+  nightly:
+    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
+
+          cat Cargo.toml
+          cat pyproject.toml
+
+      - name: Build Wheels
+        run: |
+          set -eux
+
+          VERSIONS=(
+            "3.9"
+            "3.10"
+            "3.11"
+            "3.12"
+            "3.13"
+          )
+
+          for version in "${VERSIONS[@]}"; do
+            docker run --rm -v $(pwd):/io -t torchft-maturin build --release --out dist --interpreter "$version"
+          done
+
+      - name: Twine Check
+        run: twine check --strict dist/*
+
+      - name: Upload to Pypi
+        run: twine upload --skip-existing dist/*
+        env:
+          TWINE_USERNAME: __token__
+          TWINE_PASSWORD: ${{ secrets.NIGHTLY_PYPI_TOKEN }}

.gitignore

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

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "torchft"
-version = "0.1.0"
+version = "0.1.1"
 edition = "2021"
 
 [dependencies]

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,29 @@
+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)
+
+with open(PYPROJECT_FILE, "w") as f:
+    toml.dump(pyproject, f)