Sometimes PyCharm run pytest on my application which I don't want. For example, I have two methods:
def test(model, x, y):
Then In my main function, even if I only. want to run 'train', PyCharm still run pytest for unit tests.
if __name__ == '__main__':
The message is below:
/Users/min/nlp/kg/myenv/bin/python /Applications/PyCharm.app/Contents/plugins/python/helpers/pycharm/_jb_pytest_runner.py --path /Users/min/nlp/kg/kg/embedding/experiments/ogb_node_classification.py
Testing started at 12:49 PM ...
Launching pytest with arguments /Users/min/nlp/kg/kg/embedding/experiments/ogb_node_classification.py --no-header --no-summary -q in /Users/min/nlp/kg/kg/embedding/experiments
============================= test session starts ==============================
The test with @torch.no_grad() is not for unit test, just a regular function like 'train'. How to prevent PyCharm from running as pytest always when I only want to run the function 'train'?