timing

mypy_primer/globals.py

@@ -185,7 +185,7 @@ def parse_options(argv: list[str]) -> _Args:
     primer_group.add_argument(
         "-j",
         "--concurrency",
-        default=multiprocessing.cpu_count(),
+        default=multiprocessing.cpu_count() - 8,
         type=int,
         help="number of subprocesses to use at a time",
     )

mypy_primer/main.py

@@ -229,18 +229,23 @@ async def inner(project: Project) -> tuple[float, Project]:
         result = await project.run_typechecker(type_checker_exe, typeshed_dir=None)
         return (result.runtime, project)
 
-    projects = select_projects()
-    results = []
-    for fut in asyncio.as_completed([inner(project) for project in projects]):
-        time_taken, project = await fut
-        results.append((time_taken, project))
-        print(f"[{len(results)}/{len(projects)}] {time_taken:6.2f}s  {project.location}")
+    import collections, statistics, random
 
-    results.sort(reverse=True)
+    projects = select_projects()
+    results = collections.defaultdict(list)
+    for _ in range(5):
+        random.shuffle(projects)
+        for i, fut in enumerate(asyncio.as_completed([inner(project) for project in projects])):
+            time_taken, project = await fut
+            results[project.location].append(time_taken)
+            print(f"[{i}/{len(projects)}] {time_taken:6.2f}s  {project.location}")
+
+    results = [(statistics.mean(times), location) for location, times in results.items()]
+    results.sort(key=lambda x: x[0], reverse=True)
     print("\n" * 5)
     print("Results:")
-    for time_taken, project in results:
-        print(f"{time_taken:6.2f}s  {project.location}")
+    for time_taken, location in results:
+        print(f"{time_taken:6.2f}s  {location}")
 
 
 # ==============================

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