split release.yml github action out of test.yml

update pre-commit hooks

.github/workflows/paper.yml

@@ -22,7 +22,7 @@ jobs:
           paper-path: paper/paper.md
 
       - name: Upload
-        uses: actions/upload-artifact@v1
+        uses: actions/upload-artifact@v4
         with:
           name: paper
           # This is the output path where Pandoc will write the compiled PDF.

.github/workflows/release.yml

@@ -0,0 +1,33 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  tests:
+    uses: ./.github/workflows/test.yml
+
+  release:
+    runs-on: ubuntu-latest
+    needs: tests
+    if: needs.tests.result == 'success'
+    permissions:
+      id-token: write
+    steps:
+      - name: Check out repo
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.10"
+
+      - name: Build and upload dist
+        run: |
+          pip install build twine
+          python -m build
+          twine check dist/* --strict
+
+      - name: Publish package distributions to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

.github/workflows/test.yml

@@ -9,13 +9,15 @@ on:
     paths: ["**/*.py", ".github/workflows/test.yml"]
   release:
     types: [published]
+  workflow_dispatch:
+  workflow_call:
 
 jobs:
   tests:
     strategy:
       matrix:
         os: [ubuntu-latest, windows-latest]
-    uses: janosh/workflows/.github/workflows/pytest-release.yml@main
+    uses: janosh/workflows/.github/workflows/pytest.yml@main
     with:
       os: ${{ matrix.os }}
       install-cmd: pip install .[test,excel]

.pre-commit-config.yaml

@@ -7,14 +7,14 @@ default_install_hook_types: [pre-commit, commit-msg]
 
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.4.8
+    rev: v0.7.3
     hooks:
       - id: ruff
         args: [--fix]
       - id: ruff-format
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.10.0
+    rev: v1.13.0
     hooks:
       - id: mypy
 
@@ -39,7 +39,7 @@ repos:
         args: [--check-filenames]
 
   - repo: https://github.com/kynan/nbstripout
-    rev: 0.7.1
+    rev: 0.8.0
     hooks:
       - id: nbstripout
         args: [--drop-empty-cells, --keep-output]

examples/functorch_mlp_ensemble.ipynb

@@ -164,8 +164,7 @@
     "\n",
     "    plt.scatter(*points.T, c=labels)\n",
     "    plt.gca().set(xlabel=\"x\", ylabel=\"y\", title=f\"{noise_std=}\")\n",
-    "    plt.show()\n",
-    "\n"
+    "    plt.show()"
    ]
   },
   {
@@ -313,8 +312,7 @@
     "ax.set_ylabel(\"loss\", color=\"tab:blue\")\n",
     "ax2 = ax.twinx()\n",
     "ax2.plot(list(metrics), [v[\"acc\"] for v in metrics.values()], color=\"red\")\n",
-    "ax2.set_ylabel(\"accuracy\", color=\"tab:red\")\n",
-    "\n"
+    "ax2.set_ylabel(\"accuracy\", color=\"tab:red\")"
    ]
   },
   {

examples/wandb_integration.ipynb

@@ -58,9 +58,9 @@
     "\n",
     "import pandas as pd\n",
     "import torch\n",
-    "import torch.nn as nn\n",
     "import torchvision\n",
     "import wandb\n",
+    "from torch import nn\n",
     "from tqdm import tqdm"
    ]
   },
@@ -183,8 +183,7 @@
     "    \"learning_rate\": 0.005,\n",
     "    \"dataset\": \"MNIST\",\n",
     "    \"architecture\": \"CNN\",\n",
-    "}\n",
-    "\n"
+    "}"
    ]
   },
   {
@@ -282,8 +281,7 @@
     "        out = self.layer1(x)\n",
     "        out = self.layer2(out)\n",
     "        out = out.reshape(out.size(0), -1)\n",
-    "        return self.fc(out)\n",
-    "\n"
+    "        return self.fc(out)"
    ]
   },
   {
@@ -373,8 +371,7 @@
     "                wandb.log(\n",
     "                    {\"epoch\": epoch, \"loss\": loss, \"accuracy\": accuracy},\n",
     "                    step=sample_count,\n",
-    "                )\n",
-    "\n"
+    "                )"
    ]
   },
   {
@@ -520,8 +517,7 @@
     "\n",
     "    wandb.finish()\n",
     "\n",
-    "    return model\n",
-    "\n"
+    "    return model"
    ]
   },
   {
@@ -657,7 +653,7 @@
     }
    ],
    "source": [
-    "timestamp = f\"{datetime.now():%Y-%m-%d@%H:%M}\"\n",
+    "timestamp = f\"{datetime.now().astimezone():%Y-%m-%d@%H:%M}\"\n",
     "config[\"ensemble_id\"] = timestamp\n",
     "# Build and train ensemble of models\n",
     "n_models = 5\n",
@@ -707,7 +703,7 @@
     }
    ],
    "source": [
-    "today = f\"{datetime.now():%Y-%m-%d}\"\n",
+    "today = f\"{datetime.now().astimezone():%Y-%m-%d}\"\n",
     "\n",
     "api = wandb.Api()\n",
     "runs = api.runs(\n",

paper/references.bib

@@ -37,12 +37,12 @@ @online{fey_fast_2019
   author = {Fey, Matthias and Lenssen, Jan Eric},
   date = {2019-04-25},
   eprint = {1903.02428},
-  eprinttype = {arxiv},
+  eprinttype = {arXiv},
   eprintclass = {cs, stat},
   doi = {10.48550/arXiv.1903.02428},
   url = {http://arxiv.org/abs/1903.02428},
   urldate = {2022-08-27},
-  pubstate = {preprint},
+  pubstate = {prepublished},
   keywords = {Computer Science - Machine Learning,Statistics - Machine Learning}
 }
 
@@ -86,7 +86,7 @@ @unpublished{kendall_what_2017
   author = {Kendall, Alex and Gal, Yarin},
   date = {2017-10-05},
   eprint = {1703.04977},
-  eprinttype = {arxiv},
+  eprinttype = {arXiv},
   eprintclass = {cs},
   url = {http://arxiv.org/abs/1703.04977},
   urldate = {2020-08-31},
@@ -113,7 +113,7 @@ @unpublished{lakshminarayanan_simple_2016
   author = {Lakshminarayanan, Balaji and Pritzel, Alexander and Blundell, Charles},
   date = {2016-12-05},
   eprint = {1612.01474},
-  eprinttype = {arxiv},
+  eprinttype = {arXiv},
   eprintclass = {cs, stat},
   url = {http://arxiv.org/abs/1612.01474},
   urldate = {2019-03-03},
@@ -126,11 +126,11 @@ @online{paszke_pytorch_2019
   author = {Paszke, Adam and Gross, Sam and Massa, Francisco and Lerer, Adam and Bradbury, James and Chanan, Gregory and Killeen, Trevor and Lin, Zeming and Gimelshein, Natalia and Antiga, Luca and Desmaison, Alban and Köpf, Andreas and Yang, Edward and DeVito, Zach and Raison, Martin and Tejani, Alykhan and Chilamkurthy, Sasank and Steiner, Benoit and Fang, Lu and Bai, Junjie and Chintala, Soumith},
   date = {2019-12-03},
   eprint = {1912.01703},
-  eprinttype = {arxiv},
+  eprinttype = {arXiv},
   eprintclass = {cs, stat},
   doi = {10.48550/arXiv.1912.01703},
   url = {http://arxiv.org/abs/1912.01703},
   urldate = {2022-08-27},
-  pubstate = {preprint},
+  pubstate = {prepublished},
   keywords = {Computer Science - Machine Learning,Computer Science - Mathematical Software,Statistics - Machine Learning}
 }