Support mypy v1 (#31)

pfnet · Feb 20, 2023 · 7d88ecc · 7d88ecc
1 parent 75619d9
commit 7d88ecc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pysen/ext/mypy_wrapper.py b/pysen/ext/mypy_wrapper.py
@@ -190,9 +190,9 @@ class MypyTarget:
 @functools.lru_cache(1)
 def _check_mypy_version() -> None:
     version = get_version("mypy")
-    if version.major != 0 or version.minor < 770:
+    if version.major >= 2 or version.major == 0 and version.minor < 770:
         raise IncompatibleVersionError(
-            f"pysen only supports mypy version >=0.770, <1. "
+            f"pysen only supports mypy version >=0.770, <2. "
             f"version {version} is not supported."
         )