Skip to content

Port NMS tests to use pytest #3852

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
May 17, 2021
Merged

Port NMS tests to use pytest #3852

merged 4 commits into from
May 17, 2021

Conversation

NicolasHug
Copy link
Member

@NicolasHug NicolasHug commented May 17, 2021

This PR ports the current tests for nms so that they rely on pytest.

It also introduces the doesnt_need_cuda decorator which is pretty much the opposite of needs_cuda: it allows not to run the CPU-only tests on CI GPU machines, since those CPU tests will be run by the CPU-only machines already. Would no_cuda be a better name? cpu_only?

@datumbox
Copy link
Contributor

+1 on the cpu_only suggestion. It took me few sec to consider what doesnt_need_cuda means. I think the first option is clearer.

@NicolasHug
Copy link
Member Author

Thanks for the feedback, I renamed it to cpu_only then

Copy link
Contributor

@datumbox datumbox left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

@NicolasHug NicolasHug merged commit 8fb76e8 into pytorch:master May 17, 2021
facebook-github-bot pushed a commit that referenced this pull request May 19, 2021
…rator (#3852)

Reviewed By: cpuhrsch

Differential Revision: D28538772

fbshipit-source-id: 208d3c67d24f4bc7d804bcc3dd56fb9be346880a
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants