Skip to content

Commit

Permalink
debug windows
Browse files Browse the repository at this point in the history
  • Loading branch information
qgallouedec committed Nov 21, 2024
1 parent 9e7a068 commit fa5bafe
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 4 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ jobs:
- name: Install dependencies
run: |
python -m pip install --upgrade pip
python -m pip install ".[dev]"
python -m pip install ".[mergekit]"
- name: Test with pytest
run: |
make test
Expand Down Expand Up @@ -88,7 +88,7 @@ jobs:
python -m pip install -U git+https://github.com/huggingface/accelerate.git
python -m pip install -U git+https://github.com/huggingface/datasets.git
python -m pip install -U git+https://github.com/huggingface/transformers.git
python -m pip install ".[dev]"
python -m pip install ".[mergekit]"
- name: Test with pytest
run: |
make test
Expand Down Expand Up @@ -149,7 +149,7 @@ jobs:
python -m pip install accelerate==0.34.0
python -m pip install datasets==2.21.0
python -m pip install transformers==4.46.0
python -m pip install ".[dev]"
python -m pip install ".[mergekit]"
- name: Test with pytest
run: |
make test
Expand Down
2 changes: 1 addition & 1 deletion Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ dev:
ln -s `pwd`/examples/scripts/ `pwd`/trl/commands

test:
python -m pytest -n auto --dist=loadfile -s -v --reruns 5 --reruns-delay 1 --only-rerun '(OSError|Timeout|HTTPError.*502|HTTPError.*504||not less than or equal to 0.01)' ./tests/
python -m pytest tests/test_callbacks.py::MergeModelCallbackTester::test_callback

precommit:
pre-commit run --all-files
Expand Down
2 changes: 2 additions & 0 deletions tests/test_callbacks.py
Original file line number Diff line number Diff line change
Expand Up @@ -299,6 +299,8 @@ def test_callback(self):
trainer.train()
last_checkpoint = get_last_checkpoint(tmp_dir)
merged_path = os.path.join(last_checkpoint, "merged")
import warnings
warnings.warn(f"merged_path: {merged_path}")
self.assertTrue(os.path.isdir(merged_path), "Merged folder does not exist in the last checkpoint.")

def test_every_checkpoint(self):
Expand Down

0 comments on commit fa5bafe

Please sign in to comment.