Add database models

.env.example

@@ -0,0 +1,15 @@
+## Copy this file to `.env` and fill in the values as needed.
+
+# Local development database (default if not set)
+DATABASE_URL=sqlite:///policyengine.db
+
+# PolicyEngine live database connection pieces (used when --db-location policyengine)
+# The CLI composes the URL as postgresql+psycopg2://... with sslmode=require by default.
+POLICYENGINE_DB_PASSWORD=
+POLICYENGINE_DB_USER=postgres
+POLICYENGINE_DB_HOST=db.usugnrssspkdutcjeevk.supabase.co
+POLICYENGINE_DB_PORT=5432
+POLICYENGINE_DB_NAME=postgres
+
+# Optional: Hugging Face token for private repos when seeding datasets from HF
+HUGGING_FACE_TOKEN=

.github/fetch_version.py

@@ -3,7 +3,6 @@ def fetch_version():
         import importlib
 
         return importlib.import_module("policyengine").__version__
-        return version
     except Exception as e:
         print(f"Error fetching version: {e}")
         return None

.github/workflows/code_changes.yaml

@@ -1,14 +1,16 @@
-# Workflow that runs on code changes to a pull request.
+# Workflow that runs on code changes to the master branch.
 
 name: Code changes
 on:
-  pull_request:
+  push:
     branches:
       - main
 
     paths:
-      - policyengine/**
+      - src/**
       - tests/**
+      - .github/**
+  workflow_dispatch:
 
 jobs:
   Lint:
@@ -20,27 +22,28 @@ jobs:
           with:
               args: ". -l 79 --check"
   Test:
-      runs-on: ubuntu-latest
+      runs-on: macos-latest
       permissions:
         contents: "read"
         id-token: "write"
       steps:
           - name: Checkout repo
-            uses: actions/checkout@v2
+            uses: actions/checkout@v4
           - name: Install uv
             uses: astral-sh/setup-uv@v5
 
           - name: Set up Python
-            uses: actions/setup-python@v2
+            uses: actions/setup-python@v5
             with:
                 python-version: '3.13'
-          - uses: "google-github-actions/auth@v2"
-            with:
-              workload_identity_provider: "projects/322898545428/locations/global/workloadIdentityPools/policyengine-research-id-pool/providers/prod-github-provider"
-              service_account: "policyengine-research@policyengine-research.iam.gserviceaccount.com"
 
           - name: Install package
-            run: uv pip install .[dev] --system
-
-          - name: Run tests
-            run: make test
+            run: uv pip install -e .[dev] --system
+          - name: Install JB
+            run: uv pip install "jupyter-book>=2.0.0a0" --system
+          - name: UV sync
+            run: uv sync
+          - name: Run tests with coverage
+            run: make test
+            env:
+              HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}

.github/workflows/deploy.yml

@@ -0,0 +1,49 @@
+# This file was created automatically with `myst init --gh-pages` 🪄 💚
+# Ensure your GitHub Pages settings for this repository are set to deploy with **GitHub Actions**.
+
+name: Deploy documentation
+on:
+  push:
+    # Runs on pushes targeting the default branch
+    branches: [master]
+env:
+  # `BASE_URL` determines, relative to the root of the domain, the URL that your site is served from.
+  # E.g., if your site lives at `https://mydomain.org/myproject`, set `BASE_URL=/myproject`.
+  # If, instead, your site lives at the root of the domain, at `https://mydomain.org`, set `BASE_URL=''`.
+  BASE_URL: /${{ github.event.repository.name }}
+
+# Sets permissions of the GITHUB_TOKEN to allow deployment to GitHub Pages
+permissions: 
+  contents: read
+  pages: write
+  id-token: write
+# Allow only one concurrent deployment, skipping runs queued between the run in-progress and latest queued.
+# However, do NOT cancel in-progress runs as we want to allow these production deployments to complete.
+concurrency:
+  group: 'pages'
+  cancel-in-progress: false
+jobs:
+  deploy:
+    name: Deploy to GitHub Pages
+    environment:
+      name: github-pages
+      url: ${{ steps.deployment.outputs.page_url }}
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Setup Pages
+        uses: actions/configure-pages@v3
+      - uses: actions/setup-node@v4
+        with:
+          node-version: 18.x
+      - name: Install MyST
+        run: npm install -g mystmd
+      - name: Build HTML Assets
+        run: cd docs && myst build --html
+      - name: Upload artifact
+        uses: actions/upload-pages-artifact@v3
+        with:
+          path: './docs/_build/html'
+      - name: Deploy to GitHub Pages
+        id: deployment
+        uses: actions/deploy-pages@v4

.github/workflows/pr_code_changes.yaml

@@ -0,0 +1,48 @@
+# Workflow that runs on code changes to a pull request.
+
+name: Code changes
+on:
+  pull_request:
+    branches:
+      - main
+
+    paths:
+      - src/**
+      - tests/**
+      - .github/**
+  workflow_dispatch:
+
+jobs:
+  Lint:
+    runs-on: ubuntu-latest
+    steps:
+        - uses: actions/checkout@v4
+        # Check formatting with ruff
+        - name: Install ruff
+          run: pip install ruff
+
+        - name: Run ruff check
+          run: ruff check .
+  Test:
+    runs-on: macos-latest
+    steps:
+        - name: Checkout repo
+          uses: actions/checkout@v4
+        - name: Install uv
+          uses: astral-sh/setup-uv@v5
+
+        - name: Set up Python
+          uses: actions/setup-python@v5
+          with:
+              python-version: '3.13'
+
+        - name: Install package
+          run: uv pip install -e .[dev] --system
+        - name: Install policyengine
+          run: uv pip install policyengine --system
+        - name: UV sync
+          run: uv sync
+        - name: Run tests with coverage
+          run: make test
+          env:
+            HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}

.github/workflows/pr_docs_changes.yaml

@@ -0,0 +1,36 @@
+# Workflow that runs on code changes to a pull request.
+
+name: Docs changes
+on:
+  pull_request:
+    branches:
+      - main
+
+    paths:
+      - docs/**
+      - .github/**
+  workflow_dispatch:
+
+jobs:
+  Test:
+      runs-on: ubuntu-latest
+      name: Test documentation builds
+      steps:
+          - name: Checkout repo
+            uses: actions/checkout@v4
+          - name: Install uv
+            uses: astral-sh/setup-uv@v5
+
+          - name: Set up Python
+            uses: actions/setup-python@v5
+            with:
+                python-version: '3.13'
+
+          - name: Install package
+            run: uv pip install -e .[dev] --system
+          - name: Install policyengine
+            run: uv pip install policyengine --system
+          - name: Install JB
+            run: uv pip install "jupyter-book>=2.0.0a0" --system
+          - name: Test documentation builds
+            run: cd docs && jupyter book build