Skip to content

Commit 7948858

Browse files
authored
trying sklearn deprecated flag
1 parent b8d49a2 commit 7948858

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Dockerfile

+1
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ RUN yes yes | ./FeTS_${VERSION}_Installer.bin --target ./FeTS_${VERSION} -- --cu
2222

2323
ENV PATH=/workspace/FeTS_${VERSION}/squashfs-root/usr/bin/:$PATH
2424
ENV LD_LIBRARY_PATH=/workspace/FeTS_${VERSION}/squashfs-root/usr/lib/:$LD_LIBRARY_PATH
25+
ENV SKLEARN_ALLOW_DEPRECATED_SKLEARN_PACKAGE_INSTALL="True"
2526

2627
# set up environment and install correct version of pytorch
2728
RUN cd ./FeTS_${VERSION}/squashfs-root/usr/bin/OpenFederatedLearning && \

0 commit comments

Comments
 (0)