diff --git a/docker/Dockerfile.torchsig b/docker/Dockerfile.torchsig index 77670121..94ab31a7 100644 --- 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"