rust tests

.github/workflows/launch_infrastructure.yaml

@@ -49,16 +49,12 @@ jobs:
         with:
           role-to-assume: ${{ secrets.AWS_IAM_INFRASTRUCTURE_ROLE_ARN }}
           aws-region: ${{ secrets.AWS_REGION }}
+      - name: Set up Docker Buildx
+        if: steps.changes.outputs.service == 'true' || github.event_name == 'schedule'
+        uses: docker/setup-buildx-action@v3
       - name: Install Flox
         if: steps.changes.outputs.service == 'true' || github.event_name == 'schedule'
         uses: flox/install-flox-action@v2
-      - name: Conditionally download artifacts
-        if: (steps.changes.outputs.service == 'true' || github.event_name == 'schedule') && matrix.service == 'equitypricemodel'
-        uses: flox/activate-action@v1
-        env:
-          AWS_S3_ARTIFACTS_BUCKET_NAME: ${{ secrets.AWS_S3_ARTIFACTS_BUCKET_NAME }}
-        with:
-          command: mask models artifacts download equitypricemodel
       - name: Build ${{ matrix.service }} image
         if: steps.changes.outputs.service == 'true' || github.event_name == 'schedule'
         uses: flox/activate-action@v1

.github/workflows/run_code_checks.yaml

@@ -27,6 +27,8 @@ jobs:
         with:
           workspaces: .
           cache-on-failure: false
+          shared-key: rust-continuous-integration
+          save-if: ${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }}
       - name: Run Rust code checks
         uses: flox/activate-action@v1
         with:
@@ -88,5 +90,4 @@ jobs:
         uses: coverallsapp/github-action@v2
         with:
           github-token: ${{ secrets.GITHUB_TOKEN }}
-          # Rust is being excluded from coverage in favor of DataFrame correctness checks
-          files: .coverage_output/python.xml
+          files: .coverage_output/python.xml .coverage_output/rust.xml

.gitignore

@@ -10,7 +10,7 @@ __pycache__/
 .envrc
 .env.nu
 .coverage_output/
-.coverage
+.coverage*
 *.csv
 *.egg-info/
 wandb/
@@ -22,3 +22,4 @@ data/
 **/*.json
 .claude/tasks/
 .scratchpad/
+

CLAUDE.md

@@ -31,7 +31,7 @@ This is a collection of guidelines and references.
 - Use Polars for [Python](https://docs.pola.rs/api/python/stable/reference/index.html) and
   [Rust](https://docs.rs/polars/latest/polars/) dataframes
 - Use `typing` module `cast` function for `tinygrad` method outputs when necessary with union types
-- Write `pytest` functions for Python tests
+- Write `pytest` functions for Python tests using plain functions, not class-based test organization
 - Ensure Rust and Python automated test suites achieve at least 90% line or statement coverage per service or library
 - Exclude generated code, third-party code, tooling boilerplate, and anything explicitly excluded in this repository
   from test coverage calculations