Skip to content

Commit

Permalink
Re-enable certain cpu tests (#1618)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #1618

They don't seem flaky anymore.

https://github.com/pytorch/torchrec/actions/runs/7467289973/job/20322923697

 {F1267326370}

Reviewed By: PaulZhang12

Differential Revision: D52637920

fbshipit-source-id: 402eabd6fe65ad38b2f1d23780714d9b5cf09329
  • Loading branch information
henrylhtsang authored and facebook-github-bot committed Jan 10, 2024
1 parent 9bc6103 commit 93de315
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion .github/workflows/unittest_ci_cpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,8 @@ jobs:
echo "numpy succeeded"
conda install -n build_binary -y pytest
conda run -n build_binary \
python -m pytest torchrec -v -s -W ignore::pytest.PytestCollectionWarning --continue-on-collection-errors -k 'not test_sharding_gloo_cw and not test_load_state_dict_prefix and not test_load_state_dict_cw_multiple_shards' --ignore-glob=**/test_utils/
python -m pytest torchrec -v -s -W ignore::pytest.PytestCollectionWarning --continue-on-collection-errors \
--ignore-glob=**/test_utils/
echo "Starting C++ Tests"
conda install -n build_binary -y gxx_linux-64
conda run -n build_binary \
Expand Down

0 comments on commit 93de315

Please sign in to comment.