Reduce pytest blocklist (#96016)
`TestCase = object` or variations of it get switched to `TestCase = NoTest`.
unittest collects test based on subclassing unittest.TestCase, so setting TestCase = object removes it from unittest test collection. pytest collects based on name (https://docs.pytest.org/en/7.1.x/reference/reference.html#confval-python_classes) but can be told to ignore a class (bottom of https://docs.pytest.org/en/7.1.x/example/pythoncollection.html#changing-naming-conventions)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96016
Approved by: https://github.com/ZainRizvi, https://github.com/huydhn
diff --git a/test/test_mps.py b/test/test_mps.py
index 4e5e42d..6abfb45 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -22,7 +22,7 @@
from torch.nn import Parameter
from torch.testing._internal import opinfo
from torch.testing._internal.common_utils import \
- (gradcheck, gradgradcheck, run_tests, TestCase, download_file, IS_CI,
+ (gradcheck, gradgradcheck, run_tests, TestCase, download_file, IS_CI, NoTest,
TEST_WITH_UBSAN, dtype_abbrs, skipIfSlowGradcheckEnv, TEST_WITH_ASAN, suppress_warnings)
from torch.testing import make_tensor
from torch.testing._comparison import TensorLikePair
@@ -175,8 +175,8 @@
# Same logic as test_cuda.py
if not torch.backends.mps.is_available():
print('MPS not available, skipping tests', file=sys.stderr)
- TestCase = object # noqa: F811
- NNTestCase = object # noqa: F811
+ TestCase = NoTest # noqa: F811
+ NNTestCase = NoTest # noqa: F811
product_version = float('.'.join(platform.mac_ver()[0].split('.')[:2]))
