improve mypyc build

hauntsaninja · Oct 24, 2024 · d1e847e · d1e847e
1 parent 17dd6bc
commit d1e847e
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/mypy_primer/type_checker.py b/mypy_primer/type_checker.py
@@ -46,19 +46,21 @@ async def pip_install(*targets: str) -> None:
         repo_dir = await ensure_repo_at_revision(repo, mypy_dir, revision_like)
         if mypyc_compile_level is not None:
             env = os.environ.copy()
-            env["MYPYC_OPT_LEVEL"] = str(mypyc_compile_level)
-            await pip_install("typing_extensions", "mypy_extensions")
+            env["MYPY_USE_MYPYC"] = "1"
+            env["MYPYC_OPT_LEVEL"] = str(mypyc_compile_level)  # can be zero
+            await pip_install("typing_extensions", "mypy_extensions", "tomli", "types-psutil", "types-setuptools")
             await run(
-                [str(venv.python), "setup.py", "--use-mypyc", "build_ext", "--inplace"],
+                [str(venv.python), "-m", "pip", "install", ".", "--no-build-isolation"],
                 cwd=repo_dir,
                 env=env,
             )
-        targets = []
-        if editable:
-            targets.append("--editable")
-        targets.append(str(repo_dir))
-        targets.append("tomli")
-        await pip_install(*targets)
+        else:
+            targets = []
+            if editable:
+                targets.append("--editable")
+            targets.append(str(repo_dir))
+            targets.append("tomli")
+            await pip_install(*targets)
 
     with open(venv.site_packages / "primer_plugin.pth", "w") as f:
         # pth file that lets us let mypy import plugins from another venv