fix test

microsoft · Sep 13, 2017 · aad54ff · aad54ff
1 parent a3df680
commit aad54ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.travis/test.sh b/.travis/test.sh
@@ -48,7 +48,8 @@ if [[ ${TASK} == "proto" ]]; then
     cd $TRAVIS_BUILD_DIR/tests/cpp_test && ../../lightgbm config=train.conf && ../../lightgbm config=predict.conf output_result=origin.pred || exit -1
     cd $TRAVIS_BUILD_DIR && git clone https://github.com/google/protobuf && cd protobuf && ./autogen.sh && ./configure && make && sudo make install && sudo ldconfig
     cd $TRAVIS_BUILD_DIR/build && rm -rf * && cmake -DUSE_PROTO=ON .. && make lightgbm || exit -1
-    cd $TRAVIS_BUILD_DIR/tests/cpp_test && ../../lightgbm config=predict.conf output_result=proto.pred && python test.py || exit -1
+    cd $TRAVIS_BUILD_DIR/tests/cpp_test && ../../lightgbm config=train.conf model_format=proto && ../../lightgbm config=predict.conf output_result=proto.pred model_format=proto || exit -1
+    cd $TRAVIS_BUILD_DIR/tests/cpp_test && python test.py || exit -1
     exit 0
 fi