Update workflow to run all tuts on GPU
@@ -15,6 +15,8 @@ jobs:
pip install -e "."
pip install -r requirements/requirements.txt
pip install -r requirements/requirements-pytorch.txt
+ pip install -r requirements/requirements-jax.txt
+ pip install -r requirements/requirements-tf2.txt
pip install -r requirements/requirements-dev.txt\
find tutorials/future/torch/ -name '*.py' | while read f
@@ -25,3 +27,21 @@ jobs:
exit 1
fi
done
+
+ find tutorials/future/jax/ -name '*.py' | while read f
+ do
+ python $f
+ if [ $? -ne 0 ]
+ then
+ exit 1
+ fi
+ done
+
+ find tutorials/future/tf2/ -name '*.py' | while read f
+ do
+ python $f
+ if [ $? -ne 0 ]
+ then
+ exit 1
+ fi
+ done
\ No newline at end of file
