From 12622ad1144c0c388f560b7f3d15ae2692b0dfb9 Mon Sep 17 00:00:00 2001 From: Laurent Date: Thu, 3 Oct 2024 08:45:03 +0000 Subject: [PATCH] add pytest-rerunfailures optional test dependency --- pyproject.toml | 1 + requirements.lock | 13 +++++++++++++ 2 files changed, 14 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index 249a0b8..a29d5ef 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -70,6 +70,7 @@ training = [ "gitpython>=3.1.43", ] test = [ + "pytest-rerunfailures>=14.0", "diffusers>=0.26.1", "transformers>=4.35.2", "piq>=0.8.0", diff --git a/requirements.lock b/requirements.lock index 9db491e..e56d25a 100644 --- a/requirements.lock +++ b/requirements.lock @@ -67,6 +67,8 @@ dill==0.3.8 # via multiprocess docker-pycreds==0.4.0 # via wandb +exceptiongroup==1.2.2 + # via pytest filelock==3.16.1 # via datasets # via diffusers @@ -113,6 +115,8 @@ importlib-metadata==8.5.0 # via diffusers importlib-resources==6.4.5 # via swagger-spec-validator +iniconfig==2.0.0 + # via pytest isoduration==20.11.0 # via jsonschema jaxtyping==0.2.34 @@ -235,6 +239,8 @@ packaging==24.1 # via huggingface-hub # via mkdocs # via neptune + # via pytest + # via pytest-rerunfailures # via refiners # via transformers paginate==0.5.7 @@ -257,6 +263,8 @@ platformdirs==4.3.6 # via mkdocs-get-deps # via mkdocstrings # via wandb +pluggy==1.5.0 + # via pytest prodigyopt==1.0 # via refiners protobuf==5.28.2 @@ -279,6 +287,10 @@ pymdown-extensions==10.10.2 # via mkdocstrings pysocks==1.7.1 # via requests +pytest==8.3.3 + # via pytest-rerunfailures +pytest-rerunfailures==14.0 + # via refiners python-dateutil==2.9.0.post0 # via arrow # via botocore @@ -377,6 +389,7 @@ tokenizers==0.20.0 # via transformers tomli==2.0.1 # via black + # via pytest # via refiners torch==2.4.1 # via bitsandbytes