frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add missing __main__ in two unittests (#97302)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97302
Approved by: https://github.com/zou3519
This commit is contained in:
Yuxin Wu
2023-03-22 19:09:04 +00:00
committed by PyTorch MergeBot
parent 28929b1205
commit 726fc366a2
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_comparison_utils.py
View File

@ -2,7 +2,7 @@
# Owner(s): ["module: internals"]
import torch
from torch.testing._internal.common_utils import TestCase
from torch.testing._internal.common_utils import TestCase, run_tests
class TestComparisonUtils(TestCase):
def test_all_equal_no_assert(self):
@ -30,3 +30,7 @@ class TestComparisonUtils(TestCase):
with self.assertRaises(RuntimeError):
torch._assert_tensor_metadata(t, [3], [1], torch.float)
if __name__ == '__main__':
run_tests()