diff --git a/dev/infra/Dockerfile b/dev/infra/Dockerfile index d3bae836cc63..55e29148a30d 100644 --- a/dev/infra/Dockerfile +++ b/dev/infra/Dockerfile @@ -64,11 +64,11 @@ RUN Rscript -e "devtools::install_version('roxygen2', version='7.2.0', repos='ht # See more in SPARK-39735 ENV R_LIBS_SITE "/usr/local/lib/R/site-library:${R_LIBS_SITE}:/usr/lib/R/library" -RUN pypy3 -m pip install numpy 'pandas<=2.0.3' scipy coverage matplotlib -RUN python3.9 -m pip install numpy pyarrow 'pandas<=2.0.3' scipy unittest-xml-reporting plotly>=4.8 'mlflow>=2.3.1' coverage matplotlib openpyxl 'memory-profiler==0.60.0' 'scikit-learn==1.1.*' +RUN pypy3 -m pip install 'numpy==1.25.1' 'pandas<=2.0.3' scipy coverage 'matplotlib==3.7.2' +RUN python3.9 -m pip install 'numpy==1.25.1' 'pyarrow==12.0.1' 'pandas<=2.0.3' scipy unittest-xml-reporting plotly>=4.8 'mlflow>=2.3.1' coverage 'matplotlib==3.7.2' openpyxl 'memory-profiler==0.60.0' 'scikit-learn==1.1.*' # Add Python deps for Spark Connect. RUN python3.9 -m pip install 'grpcio>=1.48,<1.57' 'grpcio-status>=1.48,<1.57' 'protobuf==3.20.3' 'googleapis-common-protos==1.56.4' # Add torch as a testing dependency for TorchDistributor -RUN python3.9 -m pip install torch torchvision torcheval +RUN python3.9 -m pip install 'torch==2.0.1' 'torchvision==0.15.2' torcheval diff --git a/pom.xml b/pom.xml index fb6208777d3f..afbb2cdbd823 100644 --- a/pom.xml +++ b/pom.xml @@ -19,6 +19,7 @@ 4.0.0 + org.apache apache