Hlu/update entailment notebook running time #237
Hlu/update entailment notebook running time #237
Conversation
Check out this pull request on ReviewNB: https://app.reviewnb.com/microsoft/nlp/pull/237 You'll be able to see notebook diffs and discuss changes. Powered by ReviewNB. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this is really good
hey @hlums there is a small conflict in the conftest. One question, are you sure this piece of code is ok:
What does this yeild does? |
@miguelgfierro I actually don't know. I think it showed up in my PR because of black autoformatting. Looking at the history, @cocochrane added this. @cocochrane, can you explain this? |
I researched how to write fixtures and the yield statement essentially allows the test to run before the rest of the fixture proceeds But yes, I don't know why this is showing up in her PR because this was already merged into staging and works as expected |
Thanks! It showed up because of black auto-formatting. :) |
|
||
|
||
@pytest.mark.integration | ||
def test_entailment_multinli_bert(notebooks): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think we should add @pytest.mark.gpu
so this is executed with the gpu tests. Is there a reason why we also wanted to check this notebook on cpu?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
OK. One of the reasons is that I'm still getting familiar with the pytest marks. :)
Previously, we did have a problem that the notebook works on GPU but not on CPU, but It's now fixed and the test is covered by the unit tests.
Description
Add a QUICK_RUN flag and reference running time to the entailment notebook.
Add integration test to the entailment notebook.
If this looks good, I will do the same thing to most of the other notebooks. May not be able to do it on the QnA notebook and sentence similarity notebook because it may be hard to subset the data in those notebooks. Need to take a closer look to decide.
Related Issues
#213
Checklist: