Add deps

Author: marvinsxtr

.github/workflows/docker-publish.yaml

@@ -32,9 +32,6 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v4
 
-      - name: Clean up disk space
-        uses: jlumbroso/free-disk-space@main
-
       # Install the cosign tool except on PR
       # https://github.com/sigstore/cosign-installer
       - name: Install cosign
@@ -94,16 +91,3 @@ jobs:
         # This step uses the identity token to provision an ephemeral certificate
         # against the sigstore community Fulcio instance.
         run: echo "${TAGS}" | xargs -I {} cosign sign --yes {}@${DIGEST}
-
-      # Build and push .sif files for Apptainer
-      - name: Clean up disk space
-        uses: jlumbroso/free-disk-space@main
-      - name: Setup Apptainer
-        uses: eWaterCycle/setup-apptainer@v2
-      - name: Build and push Apptainer
-        env:
-          TAGS: ${{ steps.meta.outputs.tags }}
-        run: |
-           echo ${{ secrets.GITHUB_TOKEN }} | apptainer registry login -u ${{ secrets.GHCR_USERNAME }} --password-stdin docker://ghcr.io
-           apptainer build container.sif docker://${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:latest
-           echo "${TAGS}" | xargs -I {} apptainer push container.sif oras://{}-sif

pyproject.toml

@@ -17,9 +17,11 @@ dependencies = [
   "optax>=0.2.6",
   "pandas>=2.3.3",
   "pytest>=8.3.5",
+  "schedulefree>=1.4.1",
   "scikit-learn==1.6.1",
   "seaborn>=0.13.2",
   "tabpfn==2.2.1",
+  "torch>=2.9.1",
 ]
 
 [dependency-groups]