From 3fa16dc2fdbaba5d11eaf9f51e1e1639c7bbe44f Mon Sep 17 00:00:00 2001
From: James Braza <jamesbraza@gmail.com>
Date: Wed, 4 Sep 2024 08:36:05 -0700
Subject: [PATCH] Fixing repo tooling/CI (#12)

---
 .github/workflows/codeflash.yml | 8 +++++---
 .github/workflows/tests.yml     | 2 ++
 .pre-commit-config.yaml         | 2 +-
 pyproject.toml                  | 1 -
 4 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/codeflash.yml b/.github/workflows/codeflash.yml
index daa9aef5..e609a58b 100644
--- a/.github/workflows/codeflash.yml
+++ b/.github/workflows/codeflash.yml
@@ -37,12 +37,14 @@ jobs:
       - name: Set up uv
         if: steps.bot_check.outputs.skip_remaining_steps == 'no'
         run: curl -LsSf https://astral.sh/uv/install.sh | sh
+      - if: steps.bot_check.outputs.skip_remaining_steps == 'no'
+        run: uv sync
       - name: Run CodeFlash on fhaviary
         if: steps.bot_check.outputs.skip_remaining_steps == 'no'
-        run: codeflash
+        run: uv run codeflash
       - name: Run CodeFlash on aviary.gsm8k
         if: steps.bot_check.outputs.skip_remaining_steps == 'no'
-        run: codeflash --module-root=packages/gsm8k/src/aviary --tests-root=packages/gsm8k/tests
+        run: uv run codeflash --module-root=packages/gsm8k/src/aviary --tests-root=packages/gsm8k/tests
       - name: Run CodeFlash on aviary.hotpotqa
         if: steps.bot_check.outputs.skip_remaining_steps == 'no'
-        run: codeflash --module-root=packages/hotpotqa/src/aviary --tests-root=packages/hotpotqa/tests
+        run: uv run codeflash --module-root=packages/hotpotqa/src/aviary --tests-root=packages/hotpotqa/tests
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 0ad84423..ac4328cc 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -32,6 +32,8 @@ jobs:
           uv run refurb .
           cd ../../hotpotqa/src
           uv run refurb .
+          cd ../../..
+          uv run refurb tests packages/gsm8k/tests packages/hotpotqa/tests
       - run: uv run pylint src packages
   test:
     runs-on: large-runner
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index b2bec893..4a84fe27 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -73,7 +73,7 @@ repos:
           - httpx
           - numpy
           - pydantic~=2.0 # Match aviary pyproject.toml
-          - SQLAlchemy[aiosqlite]~=2.0 # Match aviary pyproject.toml and dev-requirements.txt
+          - SQLAlchemy[aiosqlite]~=2.0 # Match aviary pyproject.toml
           - tenacity
           - types-Pillow
           - pandas-stubs
diff --git a/pyproject.toml b/pyproject.toml
index 36d21b87..fdb7762d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -330,7 +330,6 @@ ignore = [
     "TCH003",
     "TD002",  # Don't care for TODO author
     "TD003",  # Don't care for TODO links
-    "TID252",  # Allow relative imports for packaging
     "TRY003",  # Overly pedantic
     "W505",  # Overly pedantic for Tool.__doc__
 ]