smaller.

IQTLabs · Nov 6, 2023 · a42d368 · a42d368
1 parent 10d8324
commit a42d368
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/docker/Dockerfile.torchsig b/docker/Dockerfile.torchsig
@@ -14,8 +14,6 @@ WORKDIR /root
 ENV DEBIAN_FRONTEND noninteractive
 # TODO: find a better way cherrypick just Torchsig itself, without Torch et al. Torchsig transforms have dependencies on
 # Torch, even though we don't need Torch for the standalone transforms we want.
-COPY --from=torchsig-builder /usr/local/lib/python3.10/dist-packages/torchsig /usr/local/lib/python3.10/dist-packages/torchsig
-COPY --from=torchsig-builder /usr/local/lib/python3.10/dist-packages/torch /usr/local/lib/python3.10/dist-packages/torch
-RUN apt-get update && apt-get install -y --no-install-recommends python3-pip && pip3 install PyWavelets numba && apt-get purge -y  python3-pip
+COPY --from=torchsig-builder /usr/local/lib/python3.10/dist-packages /usr/local/lib/python3.10/dist-packages
 RUN python3 -c "from torchsig.transforms import transforms"
 RUN python3 -c "from gamutrf import grscan"