add a limit to the download loop

This commit is contained in:
Daniel Gatis 2024-08-26 17:41:00 -03:00
parent 02695f5bf2
commit 95b81143c9

View File

@ -19,6 +19,13 @@ jobs:
run: pip install .[cli,dev]
- name: Test installation with pytest
run: |
rembg d
while [ $? -ne 0 ]; do !!; done
attempt=0
until rembg d || [ $attempt -eq 5 ]; do
attempt=$((attempt+1))
echo "Attempt $attempt to download the models..."
done
if [ $attempt -eq 5 ]; then
echo "downloading the models failed 5 times, exiting..."
exit 1
fi
pytest