fix: uv pip install to use --platform arg

docker/Dockerfile

@@ -13,7 +13,17 @@ RUN apt-get update \
 
 COPY --from=ghcr.io/astral-sh/uv:latest /uv /bin/uv
 COPY . .
-RUN chmod +x entrypoints.sh \
-&& uv pip install --no-cache -e . --find-links https://download.pytorch.org/whl/torch_stable.html
+RUN chmod +x entrypoints.sh
+
+ARG TARGETPLATFORM
+
+RUN echo "Building for TARGETPLATFORM: $TARGETPLATFORM"
+
+# jank as fuck because pytorch has different versions for cpu for darwin VS linux, see pyproject.toml for specifics
+RUN if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
+    uv pip install --no-cache -e . --find-links https://download.pytorch.org/whl/torch_stable.html; \
+else \
+    uv pip install --no-cache -e .; \
+fi
 
 ENTRYPOINT ["./entrypoints.sh"]