python-gfm 0.1.4__tar.gz → 0.1.5__tar.gz

{python_gfm-0.1.4 → python_gfm-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-gfm
-Version: 0.1.4
+Version: 0.1.5
 Summary: An AI copilot for graph data and models (Under active development).
 Author-email: BUAA SKLCCSE <your.email@example.com>
 License: Apache-2.0

{python_gfm-0.1.4 → python_gfm-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "python-gfm"
-version = "0.1.4"
+version = "0.1.5"
 authors = [
   { name = "BUAA SKLCCSE", email = "your.email@example.com" },
 ]
@@ -55,4 +55,8 @@ gfm-sa2gfm-downstream = "pygfm.cli.sa2gfm:downstream_main"
 
 [tool.setuptools.packages.find]
 where = ["src", "scripts", "ckpts"]
-include = ["pygfm*"]
+include = ["pygfm*"]
+
+# Ship scripts/ as a zip next to pygfm so ``pip install`` can run YAML without a git checkout.
+[tool.setuptools.package-data]
+pygfm = ["_scripts_bundle.zip"]

{python_gfm-0.1.4 → python_gfm-0.1.5}/python_gfm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-gfm
-Version: 0.1.4
+Version: 0.1.5
 Summary: An AI copilot for graph data and models (Under active development).
 Author-email: BUAA SKLCCSE <your.email@example.com>
 License: Apache-2.0

{python_gfm-0.1.4 → python_gfm-0.1.5}/python_gfm.egg-info/SOURCES.txt

@@ -7,6 +7,8 @@ python_gfm.egg-info/entry_points.txt
 python_gfm.egg-info/requires.txt
 python_gfm.egg-info/top_level.txt
 src/pygfm/__init__.py
+src/pygfm/__main__.py
+src/pygfm/_scripts_bundle.zip
 src/pygfm/baseline_models/__init__.py
 src/pygfm/baseline_models/bim_gfm/__init__.py
 src/pygfm/baseline_models/bim_gfm/ablation.py
@@ -367,6 +369,7 @@ src/pygfm/cli/run_yaml.py
 src/pygfm/cli/sa2gfm.py
 src/pygfm/cli/script_runner.py
 src/pygfm/cli/baselines/__init__.py
+src/pygfm/cli/baselines/mdgpt.py
 src/pygfm/cli/baselines/stub_config.py
 src/pygfm/private/__init__.py
 src/pygfm/private/core/__init__.py

python_gfm-0.1.5/src/pygfm/__main__.py

@@ -0,0 +1,6 @@
+"""Support ``python -m pygfm -c config.yaml`` (same as the ``pygfm`` / ``gfm`` entry points)."""
+
+from pygfm.cli.run_yaml import main
+
+if __name__ == "__main__":
+    main()

python_gfm-0.1.5/src/pygfm/cli/baselines/mdgpt.py

@@ -0,0 +1,14 @@
+"""MDGPT YAML runners that do not require a repo ``scripts/`` checkout (pip-friendly).
+
+Overrides script-based stubs from ``baseline_registry`` for pretrain / finetune.
+``finetune_graph`` still uses ``scripts/mdgpt/finetune_graph.py`` when ``PYGFM_REPO_ROOT`` is set.
+"""
+
+from __future__ import annotations
+
+from pygfm.cli.mdgpt_stages import run_mdgpt_finetune, run_mdgpt_pretrain
+
+RUNNERS = {
+    "pretrain": run_mdgpt_pretrain,
+    "finetune": run_mdgpt_finetune,
+}

{python_gfm-0.1.4 → python_gfm-0.1.5}/src/pygfm/cli/script_runner.py

@@ -2,8 +2,12 @@
 Run repository ``scripts/<baseline>/*.py`` like the original CLI: write a flat YAML and pass
 ``-c`` (for parsers that support it), then call ``main()``.
 
-Requires a repo checkout with ``scripts/``, or set ``PYGFM_REPO_ROOT`` to the repository root
-(the directory that contains ``scripts/`` and ``src/``).
+Resolution order for ``scripts/``:
+
+1. ``PYGFM_REPO_ROOT`` / ``PYGFM_SCRIPTS_ROOT``
+2. A git checkout (walk parents from this file, or ``cwd`` with ``./scripts``)
+3. **Wheel / pip install:** ``pygfm._scripts_bundle.zip`` next to this package — unpacked once under
+   the system temp dir (same layout as repo ``scripts/``).
 
 Some scripts only expose logic under ``if __name__ == "__main__"``; those need one-off handling
 (see ``_run_sa2gfm_script``).
@@ -11,10 +15,12 @@ Some scripts only expose logic under ``if __name__ == "__main__"``; those need o
 
 from __future__ import annotations
 
+import hashlib
 import importlib.util
 import os
 import sys
 import tempfile
+import zipfile
 from pathlib import Path
 from typing import Any, Callable
 
@@ -54,6 +60,34 @@ DEFAULT_STAGE_FILES: dict[str, str] = {
 }
 
 
+def _bundled_scripts_dir() -> Path | None:
+    """Unpack ``_scripts_bundle.zip`` (shipped in the wheel) to a temp dir; return ``.../scripts``."""
+    try:
+        import pygfm
+    except ImportError:
+        return None
+    pkg_root = Path(pygfm.__file__).resolve().parent
+    zpath = pkg_root / "_scripts_bundle.zip"
+    if not zpath.is_file():
+        return None
+    key = hashlib.sha256(zpath.read_bytes()).hexdigest()[:24]
+    cache = Path(tempfile.gettempdir()) / "pygfm_bundled_scripts" / key
+    scripts_dir = cache / "scripts"
+    done = cache / ".extracted"
+    if not done.is_file():
+        import shutil
+
+        if cache.exists():
+            shutil.rmtree(cache, ignore_errors=True)
+        cache.mkdir(parents=True, exist_ok=True)
+        with zipfile.ZipFile(zpath) as zf:
+            zf.extractall(cache)
+        done.write_text("1", encoding="ascii")
+    if not (scripts_dir / "mdgpt").is_dir():
+        return None
+    return scripts_dir.resolve()
+
+
 def find_scripts_root() -> Path | None:
     env = os.environ.get("PYGFM_REPO_ROOT") or os.environ.get("PYGFM_SCRIPTS_ROOT")
     if env:
@@ -72,6 +106,9 @@ def find_scripts_root() -> Path | None:
     cwd = Path.cwd()
     if (cwd / "scripts").is_dir():
         return (cwd / "scripts").resolve()
+    bundled = _bundled_scripts_dir()
+    if bundled is not None:
+        return bundled
     return None
 
 
@@ -158,8 +195,8 @@ def run_script_main(baseline: str, stage: str, cfg: dict[str, Any]) -> None:
     if path is None:
         raise FileNotFoundError(
             f"No script for baseline={baseline!r}, stage={stage!r}. "
-            "Set PYGFM_REPO_ROOT to the repo root (directory containing scripts/), "
-            "or run from that repo with a scripts/ folder."
+            "Set PYGFM_REPO_ROOT to the repo root, run from a checkout with scripts/, "
+            "or use a python-gfm wheel that includes _scripts_bundle.zip."
         )
 
     if baseline.lower() == "sa2gfm" and stage.lower() == "detect":