Add requirements for demo

CyberZHG · Aug 3, 2019 · 61faaeb · 61faaeb
1 parent 0e623f3
commit 61faaeb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/demo/CoLA/requirements.txt b/demo/CoLA/requirements.txt
@@ -0,0 +1 @@
+keras-bert
diff --git a/demo/SST-2/requirements.txt b/demo/SST-2/requirements.txt
@@ -0,0 +1 @@
+keras-bert
diff --git a/test.sh b/test.sh
@@ -1,3 +1,3 @@
 #!/usr/bin/env bash
 pycodestyle --max-line-length=120 keras_xlnet tests demo && \
-    nosetests --nocapture --with-coverage --cover-erase --cover-html --cover-html-dir=htmlcov --cover-package=keras_xlnet --with-doctest
+    nosetests --nocapture --with-coverage --cover-erase --cover-html --cover-html-dir=htmlcov --cover-package=keras_xlnet --with-doctest tests