

set -ex



pip check
python -c "import os; os.environ['KERAS_BACKEND'] = 'numpy'; import keras;"
python -c "import os; os.environ['KERAS_BACKEND'] = 'jax'; import keras;"
python -c "import os; os.environ['KERAS_BACKEND'] = 'torch'; import keras;"
export KERAS_BACKEND=torch
python -c "import keras;"
python -c "import keras.backend;"
python -c "import keras.datasets;"
python -c "import keras.layers;"
python -c "import keras.utils;"
pytest integration_tests/torch_workflow_test.py
export KERAS_BACKEND=
exit 0
