Add --prerelease=allow, add missing RUN

Author: XkunW

.github/workflows/docs.yml

@@ -70,7 +70,7 @@ jobs:
         run: uv sync --all-extras --group docs --prerelease=allow
 
       - name: Build docs
-        run: uv run mkdocs build
+        run: uv run mkdocs build --prerelease=allow
 
       - name: Create .nojekyll file
         run: touch site/.nojekyll

.github/workflows/unit_tests.yml

@@ -62,7 +62,7 @@ jobs:
 
       - name: Install dependencies and check code
         run: |
-          uv run pytest -m "not integration_test" --cov vec_inf --cov-report=xml tests
+          uv run pytest -m "not integration_test" --cov vec_inf --cov-report=xml tests --prerelease=allow
 
       - name: Install the core package only
         run: uv sync --no-dev

Dockerfile

@@ -56,7 +56,7 @@ WORKDIR /vec-inf
 COPY . /vec-inf
 
 # Install project dependencies with build requirements
-uv pip install --system -e .[dev] --prerelease=allow
+RUN uv pip install --system -e .[dev] --prerelease=allow
 
 # Install a single, system NCCL (from NVIDIA CUDA repo in base image)
 RUN apt-get update && apt-get install -y --allow-change-held-packages\