OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Enable typechecking for torch.testing._internal.common_quantized.* (#44805)

Browse Source 
Summary:
Addresses a subproblem of [Issue 42969](https://github.com/pytorch/pytorch/issues/42969)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/44805

Reviewed By: malfet

Differential Revision: D23742754

Pulled By: janeyx99

fbshipit-source-id: e916a6a0c049cac318549a485d47f19363087d15
This commit is contained in:
Jane (Yuan) Xu 2020-09-17 14:16:41 -07:00 committed by Facebook GitHub Bot
parent f5b92332c1
commit 1c996b7170
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mypy.ini
View File

@ -62,9 +62,6 @@ ignore_errors = True
[mypy-torch.testing._internal.codegen.*]
ignore_errors = True
[mypy-torch.testing._internal.common_quantized.*]
ignore_errors = True
[mypy-torch.testing._internal.jit_utils.*]
ignore_errors = True

2
torch/backends/quantized/__init__.py
View File

@ -50,3 +50,5 @@ class QuantizedEngine(types.ModuleType):
# This is the sys.modules replacement trick, see
# https://stackoverflow.com/questions/2447353/getattr-on-a-module/7668273#7668273
sys.modules[__name__] = QuantizedEngine(sys.modules[__name__], __name__)
engine: str
supported_engines: List[str]