8 add more data

.github/workflows/tests.yml

@@ -0,0 +1,91 @@
+name: Tests
+
+on:
+  push:
+    branches:
+      - "**"
+  pull_request:
+
+permissions:
+  contents: write
+
+jobs:
+  pytest:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Check out repository
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.10"
+
+      - name: Install test dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install pytest pytest-cov numpy matplotlib python-dotenv tqdm transformers
+          python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
+
+      - name: Run unit tests with coverage
+        run: |
+          pytest \
+            --cov=src/lmlm-audit \
+            --cov-report=term-missing \
+            --cov-report=xml \
+            --cov-report=json
+
+      - name: Upload coverage reports
+        uses: actions/upload-artifact@v4
+        with:
+          name: coverage-reports
+          path: |
+            coverage.xml
+            coverage.json
+
+      - name: Read coverage percentage
+        id: coverage
+        run: |
+          python - <<'PY'
+          import json
+          from pathlib import Path
+
+          data = json.loads(Path("coverage.json").read_text())
+          percent = float(data["totals"]["percent_covered"])
+          rounded = f"{percent:.0f}%"
+
+          if percent >= 90:
+              color = "brightgreen"
+          elif percent >= 80:
+              color = "green"
+          elif percent >= 70:
+              color = "yellowgreen"
+          elif percent >= 60:
+              color = "yellow"
+          elif percent >= 50:
+              color = "orange"
+          else:
+              color = "red"
+
+          with Path("/tmp/coverage_outputs.txt").open("w") as f:
+              f.write(f"message={rounded}\n")
+              f.write(f"color={color}\n")
+          PY
+          cat /tmp/coverage_outputs.txt >> "$GITHUB_OUTPUT"
+
+      - name: Generate coverage badge
+        if: github.event_name == 'push'
+        uses: emibcn/badge-action@v2
+        with:
+          label: coverage
+          status: ${{ steps.coverage.outputs.message }}
+          color: ${{ steps.coverage.outputs.color }}
+          path: badges/coverage.svg
+
+      - name: Commit coverage badge
+        if: github.event_name == 'push'
+        uses: stefanzweifel/git-auto-commit-action@v5
+        with:
+          commit_message: "chore(ci): update coverage badge"
+          file_pattern: badges/coverage.svg

.gitignore

@@ -22,8 +22,15 @@ wheels/
 # Weights & Biases
 wandb
 
-# Logging
-outputs/
-
 # uv
 uv.lock
+
+# Claude
+/.claude
+CLAUDE.md
+
+# Ruflo (Claude)
+/.swarm
+/.claude-flow
+ruvector.db
+.mcp.json

README.md

@@ -14,6 +14,10 @@
 <p align="center">
 </p>
 
+<p align="center">
+  <img src="./badges/coverage.svg" alt="Coverage badge" />
+</p>
+
 <p align="center">
   <a href="./docs/Auditing_Forgetting_in_Limited_Memory_Language_Models.pdf">Project Proposal</a>
   |