OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity

fix issue with override tests clobbering each other by qualifying with module name

Browse Source
This commit is contained in:
Nathan Goldbaum 2020-03-04 12:46:46 -06:00
parent 9ba23aa30b
commit b539fec6cc
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_overrides.py
View File

@ -481,6 +481,10 @@ def generate_tensor_like_override_tests(cls):
for func, override in get_testing_overrides().items(): for func, override in get_testing_overrides().items():
test_method = test_generator(func, override) test_method = test_generator(func, override)
module = func.__module__
if module:
name = 'test_{}_{}'.format(module.replace('.', '_'), func.__name__)
else:
name = 'test_{}'.format(func.__name__) name = 'test_{}'.format(func.__name__)
test_method.__name__ = name test_method.__name__ = name
setattr(cls, name, test_method) setattr(cls, name, test_method)