File tree Expand file tree Collapse file tree 4 files changed +16
-2
lines changed Expand file tree Collapse file tree 4 files changed +16
-2
lines changed Original file line number Diff line number Diff line change @@ -33,6 +33,12 @@ COPY model_zoo /model_zoo
33
33
RUN python -m pip install --quiet -r /model_zoo/requirements.txt \
34
34
--extra-index-url=${EXTRA_PYPI_INDEX}
35
35
36
+ # Install elasticdl_preprocessing package
37
+ COPY build/elasticdl_preprocessing-develop-py3-none-any.whl /
38
+ RUN python -m pip install --quiet /elasticdl_preprocessing-develop-py3-none-any.whl \
39
+ --extra-index-url=${EXTRA_PYPI_INDEX} \
40
+ && rm /elasticdl_preprocessing-develop-py3-none-any.whl
41
+
36
42
# Install elasticdl package
37
43
COPY build/elasticdl-develop-py3-none-any.whl /
38
44
RUN python -m pip install --quiet /elasticdl-develop-py3-none-any.whl \
Original file line number Diff line number Diff line change @@ -46,4 +46,7 @@ mv coverage.xml ./build
46
46
# Create elasticdl package
47
47
mkdir -p ./elasticdl/go/bin
48
48
cp /tmp/elasticdl_ps ./elasticdl/go/bin/
49
+ rm -rf ./build/lib
49
50
python setup.py --quiet bdist_wheel --dist-dir ./build
51
+ rm -rf ./build/lib
52
+ python setup_preprocessing.py --quiet bdist_wheel --dist-dir ./build
Original file line number Diff line number Diff line change 15
15
16
16
with open ("elasticdl/requirements.txt" ) as f :
17
17
required_deps = f .read ().splitlines ()
18
+ required_deps .append ("elasticdl_preprocessing" )
18
19
19
20
extras = {}
20
21
with open ("elasticdl/requirements-dev.txt" ) as f :
33
34
install_requires = required_deps ,
34
35
extras_require = extras ,
35
36
python_requires = ">=3.5" ,
36
- packages = find_packages (exclude = ["*test*" ]),
37
+ packages = find_packages (
38
+ exclude = ["*test*" , "elasticdl_preprocessing*" , "model_zoo*" ]
39
+ ),
37
40
package_data = {
38
41
"" : [
39
42
"proto/*.proto" ,
Original file line number Diff line number Diff line change 34
34
install_requires = required_deps ,
35
35
extras_require = extras ,
36
36
python_requires = ">=3.5" ,
37
- packages = find_packages (include = ["*elasticdl_preprocessing*" ],),
37
+ packages = find_packages (
38
+ include = ["elasticdl_preprocessing*" ], exclude = ["*test*" ]
39
+ ),
38
40
package_data = {"" : ["requirements.txt" ]},
39
41
)
You can’t perform that action at this time.
0 commit comments