mirror of
https://github.com/huggingface/trl.git
synced 2025-10-20 18:43:52 +08:00
d144e73e78
Co-authored-by: LeonEricsson <70749762+LeonEricsson@users.noreply.github.com> Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com> Co-authored-by: Quentin Gallouédec <gallouedec.quentin@gmail.com>
34 lines
1.1 KiB
Makefile
34 lines
1.1 KiB
Makefile
.PHONY: test precommit common_tests slow_tests test_examples tests_gpu test_experimental
|
|
|
|
check_dirs := examples tests trl
|
|
|
|
ACCELERATE_CONFIG_PATH = `pwd`/examples/accelerate_configs
|
|
COMMAND_FILES_PATH = `pwd`/commands
|
|
|
|
test:
|
|
pytest -n auto -m "not slow and not low_priority" -s -v --reruns 5 --reruns-delay 1 --only-rerun '(OSError|Timeout|HTTPError.*502|HTTPError.*504||not less than or equal to 0.01)' tests/
|
|
|
|
precommit:
|
|
python scripts/add_copyrights.py
|
|
pre-commit run --all-files
|
|
doc-builder style trl tests docs/source --max_len 119
|
|
|
|
slow_tests:
|
|
pytest -m "slow" tests/ $(if $(IS_GITHUB_CI),--report-log "slow_tests.log",)
|
|
|
|
test_examples:
|
|
touch temp_results_sft_tests.txt
|
|
for file in $(ACCELERATE_CONFIG_PATH)/*.yaml; do \
|
|
TRL_ACCELERATE_CONFIG=$${file} bash $(COMMAND_FILES_PATH)/run_sft.sh; \
|
|
echo $$?','$${file} >> temp_results_sft_tests.txt; \
|
|
done
|
|
|
|
touch temp_results_dpo_tests.txt
|
|
for file in $(ACCELERATE_CONFIG_PATH)/*.yaml; do \
|
|
TRL_ACCELERATE_CONFIG=$${file} bash $(COMMAND_FILES_PATH)/run_dpo.sh; \
|
|
echo $$?','$${file} >> temp_results_dpo_tests.txt; \
|
|
done
|
|
|
|
test_experimental:
|
|
pytest -k "experimental"
|