fix deps (#2)

Signed-off-by: weiwee <[email protected]>
FederatedAI · Dec 7, 2020 · 61330e3 · 61330e3
1 parent 4dfe859
commit 61330e3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/bin/run.sh b/bin/run.sh
@@ -1,7 +1,8 @@
 DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 PaddleDetection="$(dirname $DIR)/deps/PaddleDetection"
-Fedvision="$(dirname $DIR)/fedvision"
-export PYTHONPATH="$PYTHONPATH:$PaddleDetection:$Fedvision"
+Fedvision="$(dirname $DIR)"
+PaddleFL="$(dirname $DIR)/deps/PaddleFL/python"
+export PYTHONPATH="$PYTHONPATH:$PaddleDetection:$Fedvision:$PaddleFL"
 
 # todo: use yaml or json
 

diff --git a/requirements.txt b/requirements.txt
@@ -8,3 +8,4 @@ PyYAML>=5.3.1
 click==7.1.2
 paddlepaddle==1.8.5
 paddle_fl==1.1.0
+zmq # todo: remove
diff --git a/setup.py b/setup.py
@@ -27,7 +27,6 @@
     "PyYAML>=5.3.1",
     "click==7.1.2",
     "paddlepaddle==1.8.5",
-    "paddle_fl==1.1.0",
 ]
 
 setup(