Merge pull request #577 from DesmonDay/main

[PGLBox] fix some codes
PaddlePaddle · Sep 22, 2023 · 076c72c · 076c72c
2 parents c63d4e9 + f991c14
commit 076c72c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/apps/PGLBox/docker/Dockerfile b/apps/PGLBox/docker/Dockerfile
@@ -28,6 +28,8 @@ RUN cd ${HOME_WORK_DIR} && \
     pip3 install protobuf==3.20.0 -U && \
     pip3 install pgl -U && \
     pip3 install gpustat==1.0.0 -U && \
+    pip3 install requests==2.28.1 -U && \
+    pip3 install joblib==1.2.0 -U && \
     rm -rf ${PADDLE_WHL}
 
 

diff --git a/apps/PGLBox/src/util.py b/apps/PGLBox/src/util.py
@@ -282,6 +282,8 @@ def save_model(exe, model_dict, args, local_model_path, model_save_path):
         else:
             run_cmd("mv %s %s" % (local_var_save_path, model_save_path))
 
+    return 0
+
 
 def upload_embedding(args, local_embed_path):
     mode, infer_result_path = parse_path(args.infer_result_path)