mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
1022443168
Test Plan: revert-hammer
Differential Revision:
D30279364 (b004307252)
Original commit changeset: c1ed77dfe43a
fbshipit-source-id: eab50857675c51e0088391af06ec0ecb14e2347e
34 lines
853 B
Python
34 lines
853 B
Python
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from mypy.plugin import Plugin
|
|
|
|
|
|
def get_correct_mypy_version():
|
|
# there's probably a more elegant way to do this
|
|
match, = re.finditer(
|
|
r'mypy==(\d+(?:\.\d+)*)',
|
|
Path('.circleci/docker/common/install_conda.sh').read_text(),
|
|
)
|
|
version, = match.groups()
|
|
return version
|
|
|
|
|
|
def plugin(version: str):
|
|
correct_version = get_correct_mypy_version()
|
|
if version != correct_version:
|
|
print(f'''\
|
|
You are using mypy version {version}, which is not supported
|
|
in the PyTorch repo. Please switch to mypy version {correct_version}.
|
|
|
|
For example, if you installed mypy via pip, run this:
|
|
|
|
pip install mypy=={correct_version}
|
|
|
|
Or if you installed mypy via conda, run this:
|
|
|
|
conda install -c conda-forge mypy={correct_version}
|
|
''', file=sys.stderr)
|
|
return Plugin
|