Merge pull request #2 from mctigger/test-nightly

- Adds github workflows with checks
- Adds a nightly release on pull request merge
- Adds a workflow for push to pypi on release

Author: mctigger

.github/workflows/nightly.yml

@@ -0,0 +1,63 @@
+name: Python package
+
+on:
+  pull_request:
+    branches: [ "main" ]
+
+jobs:
+  # This job runs only ONCE on a single Python version.
+  formatting-check:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.12'
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install .[dev]
+    - name: Lint and Format with Ruff
+      run: |
+        ruff check .
+        ruff format --check .
+
+  # This job now tests against Python 3.9, 3.10, 3.11, and 3.12.
+  pytest-check:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v5
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install .[dev]
+    - name: Run pytest
+      run: |
+        pytest
+
+  # This job depends on both formatting and tests passing.
+  build-check:
+    runs-on: ubuntu-latest
+    needs: [formatting-check, pytest-check]
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.x"
+      - name: Install build dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install build
+      - name: Build package
+        run: python -m build

.github/workflows/pr-checks.yml

@@ -0,0 +1,71 @@
+name: Publish Nightly
+
+on:
+  push:
+    branches: ["main"]
+
+permissions:
+  contents: read
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    permissions:
+      # Grant write permission ONLY to this job to modify pyproject.toml
+      contents: write
+
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.x"
+      - name: Create dynamic nightly version
+        run: |
+          # Extract the base version from pyproject.toml (e.g., "0.5.0")
+          BASE_VERSION=$(cat pyproject.toml | grep -oP 'version = "\K[^"]+')
+
+          # Create a date-based suffix (e.g., ".dev20250719203543")
+          NIGHTLY_SUFFIX=".dev$(date +'%Y%m%d%H%M%S')"
+
+          # Combine them into the full nightly version (e.g., "0.5.0.dev20250719203543")
+          NIGHTLY_VERSION="$BASE_VERSION$NIGHTLY_SUFFIX"
+
+          echo "Setting package version to: $NIGHTLY_VERSION"
+
+          # Update the version in pyproject.toml
+          sed -i "s/version = \"$BASE_VERSION\"/version = \"$NIGHTLY_VERSION\"/" pyproject.toml
+      - name: Build release distributions
+        run: |
+          python -m pip install build
+          python -m build
+
+      - name: Upload distributions
+        uses: actions/upload-artifact@v4
+        with:
+          name: release-dists
+          path: dist/
+
+  publish:
+    runs-on: ubuntu-latest
+    needs:
+      - build
+    permissions:
+      # This permission is mandatory for trusted publishing to PyPI.
+      id-token: write
+
+    # Using a dedicated environment for PyPI is a recommended security practice.
+    environment:
+      name: pypi
+      url: https://pypi.org/p/tensorcontainer
+
+    steps:
+      - name: Retrieve release distributions
+        uses: actions/download-artifact@v4
+        with:
+          name: release-dists
+          path: dist/
+
+      - name: Publish release distributions to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          packages-dir: dist/

.github/workflows/publish-nightly.yml

@@ -0,0 +1,59 @@
+name: Publish Release
+
+# This workflow runs when a new GitHub Release is published.
+on:
+  release:
+    types: [published]
+
+jobs:
+  # This job builds the package using the version defined in pyproject.toml.
+  build:
+    runs-on: ubuntu-latest
+    permissions:
+      # This permission is needed to check out the code.
+      contents: read
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.x"
+
+      - name: Install build dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install build
+
+      - name: Build release distributions
+        run: python -m build
+
+      - name: Upload distributions as an artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: release-dists
+          path: dist/
+
+  # This job publishes the official release to PyPI.
+  publish:
+    needs: build
+    runs-on: ubuntu-latest
+    permissions:
+      # This permission is mandatory for trusted publishing to PyPI.
+      id-token: write
+
+    # You must configure this environment in your repository settings.
+    environment:
+      name: pypi
+      url: https://pypi.org/p/tensorcontainer/${{ github.event.release.tag_name }} 
+
+    steps:
+      - name: Retrieve release distributions from artifact
+        uses: actions/download-artifact@v4
+        with:
+          name: release-dists
+          path: dist/
+
+      - name: Publish package to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

.github/workflows/publish-release.yml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tensorcontainer"
-version = "0.5.0"
+version = "0.5.1"
 description = "TensorDict-like functionality for PyTorch with PyTree compatibility and torch.compile support"
 authors = [{name="Tim Joseph", email="tim@mctigger.com"}]
 license = {text = "MIT"}