Install tensorflow-docs requirements for notebook check (#183)

* Install tensorflow-docs requirements for notebook check Change-Id: I5f6724ab9b9372e878e2ff6539d644d202c0d72d * Update comment Change-Id: I1ae55acd2d15c9d2eb39e23baaaadc7adfc7c38a
quantumlib · Jul 14, 2021 · a230b6d · a230b6d
1 parent c9f26eb
commit a230b6d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dev_tools/nbfmt b/dev_tools/nbfmt
@@ -24,6 +24,12 @@ if [[ ! -f dev_tools/nbformat ]]; then
   chmod u+x dev_tools/nbformat
 fi
 
+# Check if tensorflow-docs.txt requirements file exists, if not grab it.
+if [[ ! -f dev_tools/tensorflow-docs.txt ]]; then
+  wget https://raw.githubusercontent.com/quantumlib/Cirq/master/dev_tools/requirements/deps/tensorflow-docs.txt -P dev_tools/
+  pip install -r dev_tools/tensorflow-docs.txt
+fi
+
 # Run the formatter.
 result=$(dev_tools/nbformat "$@")
 status=$?