remove fastapi and pydantic extras

.github/workflows/tests.yml

@@ -228,6 +228,7 @@ jobs:
           echo "AXOLOTL_ARGS=${{ matrix.axolotl_args}}" >> $GITHUB_ENV
           echo "AXOLOTL_EXTRAS=${{ matrix.axolotl_extras}}" >> $GITHUB_ENV
           echo "CUDA=${{ matrix.cuda }}" >> $GITHUB_ENV
+          echo "MODAL_IMAGE_BUILDER_VERSION=2024.10" >> $GITHUB_ENV
           echo "N_GPUS=${{ matrix.num_gpus }}" >> $GITHUB_ENV
       - name: Run tests job on Modal
         run: |
@@ -268,6 +269,7 @@ jobs:
           echo "AXOLOTL_ARGS=${{ matrix.axolotl_args}}" >> $GITHUB_ENV
           echo "AXOLOTL_EXTRAS=${{ matrix.axolotl_extras}}" >> $GITHUB_ENV
           echo "CUDA=${{ matrix.cuda }}" >> $GITHUB_ENV
+          echo "MODAL_IMAGE_BUILDER_VERSION=2024.10" >> $GITHUB_ENV
           echo "N_GPUS=${{ matrix.num_gpus }}" >> $GITHUB_ENV
       - name: Run tests job on Modal
         run: |

cicd/tests.py

@@ -38,16 +38,12 @@ temp_dir = tempfile.mkdtemp()
 with open(pathlib.Path(temp_dir) / "Dockerfile", "w", encoding="utf-8") as f:
     f.write(dockerfile_contents)
 
-cicd_image = (
+cicd_image = Image.from_dockerfile(
-    Image.from_dockerfile(
+    pathlib.Path(temp_dir) / "Dockerfile",
-        pathlib.Path(temp_dir) / "Dockerfile",
+    context_mount=None,
-        context_mount=None,
+    force_build=True,
-        force_build=True,
+    gpu="A10G",
-        gpu="A10G",
+).env(df_args)
-    )
-    .env(df_args)
-    .pip_install("fastapi==0.110.0", "pydantic==2.6.3")
-)
 
 app = App("Axolotl CI/CD", secrets=[])
 

docs/dataset-formats/stepwise_supervised.qmd

@@ -23,4 +23,4 @@ Here's a simple example of a stepwise supervised dataset entry:
   ],
   "labels": [true, false]
 }
-```
+```