python-package.yml

name: Python package

on:
  push:
    branches: [main]
  pull_request:
    branches: [main]

jobs:
  build:
    runs-on: ubuntu-latest
    permissions:
      contents: read
    strategy:
      matrix:
        python-version: ["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: Build source and wheel distributions
        run: |
          python -m pip install --upgrade pip build
          python -m build --sdist --wheel

      - name: Install built wheel
        run: |
          python -m pip install dist/*.whl

      - name: Import smoke test
        run: |
          cd /tmp
          python - <<'PY'
          import pymdp
          from pymdp.agent import Agent

          print("Imported pymdp from", pymdp.__file__)
          assert "site-packages" in pymdp.__file__
          print("Agent symbol:", Agent.__name__)
          PY