From d91cfe66343ff27ce605e88db144612806632ac3 Mon Sep 17 00:00:00 2001
From: qiaolongfei <qiaolongfei@baidu.com>
Date: Wed, 27 Dec 2017 15:33:16 +0800
Subject: [PATCH] update setup.py

---
 server/build.sh | 9 +++++++--
 server/setup.py | 5 ++++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/server/build.sh b/server/build.sh
index 0d7ba99d..c9c49076 100644
--- a/server/build.sh
+++ b/server/build.sh
@@ -1,2 +1,7 @@
-protoc visualdl/onnx/onnx.proto --python_out .
-python setup.py bdist_wheel
+script=$(readlink -f "$0")
+script_path=$(dirname "$script")
+
+pushd $script_path
+  protoc visualdl/onnx/onnx.proto --python_out .
+  python setup.py bdist_wheel
+popd
diff --git a/server/setup.py b/server/setup.py
index 6b8c71ed..b20a62b8 100644
--- a/server/setup.py
+++ b/server/setup.py
@@ -1,6 +1,9 @@
 from setuptools import setup
 
-packages = ['visualdl', 'visualdl.frontend.dist', 'visualdl.mock']
+packages = ['visualdl',
+            'visualdl.onnx',
+            'visualdl.mock',
+            'visualdl.frontend.dist']
 
 setup(
     name="visualdl",
-- 
GitLab