add pytest-rerunfailures optional test dependency

This commit is contained in:
Laurent 2024-10-03 08:45:03 +00:00 committed by Laureηt
parent ae5ef4f14c
commit 12622ad114
2 changed files with 14 additions and 0 deletions

View file

@ -70,6 +70,7 @@ training = [
"gitpython>=3.1.43", "gitpython>=3.1.43",
] ]
test = [ test = [
"pytest-rerunfailures>=14.0",
"diffusers>=0.26.1", "diffusers>=0.26.1",
"transformers>=4.35.2", "transformers>=4.35.2",
"piq>=0.8.0", "piq>=0.8.0",

View file

@ -67,6 +67,8 @@ dill==0.3.8
# via multiprocess # via multiprocess
docker-pycreds==0.4.0 docker-pycreds==0.4.0
# via wandb # via wandb
exceptiongroup==1.2.2
# via pytest
filelock==3.16.1 filelock==3.16.1
# via datasets # via datasets
# via diffusers # via diffusers
@ -113,6 +115,8 @@ importlib-metadata==8.5.0
# via diffusers # via diffusers
importlib-resources==6.4.5 importlib-resources==6.4.5
# via swagger-spec-validator # via swagger-spec-validator
iniconfig==2.0.0
# via pytest
isoduration==20.11.0 isoduration==20.11.0
# via jsonschema # via jsonschema
jaxtyping==0.2.34 jaxtyping==0.2.34
@ -235,6 +239,8 @@ packaging==24.1
# via huggingface-hub # via huggingface-hub
# via mkdocs # via mkdocs
# via neptune # via neptune
# via pytest
# via pytest-rerunfailures
# via refiners # via refiners
# via transformers # via transformers
paginate==0.5.7 paginate==0.5.7
@ -257,6 +263,8 @@ platformdirs==4.3.6
# via mkdocs-get-deps # via mkdocs-get-deps
# via mkdocstrings # via mkdocstrings
# via wandb # via wandb
pluggy==1.5.0
# via pytest
prodigyopt==1.0 prodigyopt==1.0
# via refiners # via refiners
protobuf==5.28.2 protobuf==5.28.2
@ -279,6 +287,10 @@ pymdown-extensions==10.10.2
# via mkdocstrings # via mkdocstrings
pysocks==1.7.1 pysocks==1.7.1
# via requests # via requests
pytest==8.3.3
# via pytest-rerunfailures
pytest-rerunfailures==14.0
# via refiners
python-dateutil==2.9.0.post0 python-dateutil==2.9.0.post0
# via arrow # via arrow
# via botocore # via botocore
@ -377,6 +389,7 @@ tokenizers==0.20.0
# via transformers # via transformers
tomli==2.0.1 tomli==2.0.1
# via black # via black
# via pytest
# via refiners # via refiners
torch==2.4.1 torch==2.4.1
# via bitsandbytes # via bitsandbytes