Expose ART Megatron installer for downstream workers

Author: Kovbo

pyproject.toml

@@ -92,6 +92,7 @@ packages = ["src/art", "src/mp_actors"]
 
 [tool.hatch.build.targets.wheel.force-include]
 ".agents/skills" = "art/skills"
+"src/art_megatron_install.py" = "art_megatron_install.py"
 
 [tool.hatch.build]
 sources = ["src"]

src/art_megatron_install.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import argparse
-import importlib.metadata as metadata
+from dataclasses import dataclass
+from importlib import metadata
 import json
 import os
 import shutil
 import subprocess
-from dataclasses import dataclass
 from urllib.parse import parse_qs
 
 from packaging.requirements import Requirement
@@ -62,17 +62,15 @@ def _megatron_requirements() -> dict[str, Requirement]:
 
 
 def _format_requirement(requirement: Requirement) -> str:
-    extras = (
-        f"[{','.join(sorted(requirement.extras))}]"
-        if requirement.extras
-        else ""
-    )
+    extras = f"[{','.join(sorted(requirement.extras))}]" if requirement.extras else ""
     if requirement.url is not None:
         return f"{requirement.name}{extras} @ {requirement.url}"
     return f"{requirement.name}{extras}{requirement.specifier}"
 
 
-def _run(command: list[str], *, env: dict[str, str] | None = None, dry_run: bool) -> None:
+def _run(
+    command: list[str], *, env: dict[str, str] | None = None, dry_run: bool
+) -> None:
     print("+", " ".join(command))
     if dry_run:
         return