clearml-truen-patch 2.1.3.post1__tar.gz

clearml_truen_patch-2.1.3.post1/MANIFEST.in

@@ -0,0 +1,2 @@
+include clearml_truen_patch.pth
+include README.md

clearml_truen_patch-2.1.3.post1/PKG-INFO

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: clearml-truen-patch
+Version: 2.1.3.post1
+Summary: Overlay patches for clearml==2.1.3 (truen). Installs into the existing clearml package so `import clearml` picks up the patches.
+Home-page: https://github.com/clearml/clearml
+Author: ClearML
+Author-email: support@clear.ml
+License: Apache License 2.0
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+Requires-Dist: clearml==2.1.3
+Dynamic: author
+Dynamic: author-email
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

clearml_truen_patch-2.1.3.post1/clearml_truen_patch/__init__.py

@@ -0,0 +1,52 @@
+""" ClearML open SDK """
+from .version import __version__
+from .task import Task
+from .model import InputModel, OutputModel, Model
+from .logger import Logger
+from .storage import StorageManager
+from .errors import UsageError
+from .datasets import Dataset
+from .hyperdatasets import (
+    HyperDataset,
+    DataView,
+    DataEntry,
+    DataSubEntry,
+    DataEntryImage,
+    DataSubEntryImage,
+    HyperDatasetManagement,
+    HyperDatasetQuery,
+)
+
+from .backend_api import browser_login  # noqa: F401
+from .automation.controller import (  # noqa: F401
+    PipelineController,  # noqa: F401
+    PipelineDecorator,  # noqa: F401
+)  # noqa: F401
+
+
+TaskTypes = Task.TaskTypes
+
+
+__all__ = [
+    "__version__",
+    "Task",
+    "TaskTypes",
+    "InputModel",
+    "OutputModel",
+    "Model",
+    "Logger",
+    "StorageManager",
+    "UsageError",
+    "Dataset",
+    "PipelineController",
+    "PipelineDecorator",
+    "browser_login",
+    "DataEntry",
+    "DataSubEntry",
+    "DataEntryImage",
+    "DataSubEntryImage",
+    "HyperDataset",
+    "HyperDatasetManagement",
+    "DataView",
+    "HyperDatasetQuery",
+]

clearml_truen_patch-2.1.3.post1/clearml_truen_patch/_autoapply.py

@@ -0,0 +1,25 @@
+"""Auto-apply overlay on interpreter startup (triggered by clearml_truen_patch.pth).
+
+Delegates to _install.needs_apply() which checks clearml version, patch
+version, and source fingerprint. Only copies files when something changed.
+"""
+import sys
+
+
+def _safe_apply():
+    try:
+        from clearml_truen_patch._install import (
+            run, needs_apply, _find_clearml_dir, _source_dir,
+        )
+        dst = _find_clearml_dir()
+        src = _source_dir()
+        if needs_apply(dst, src):
+            run(force=True)
+    except SystemExit:
+        # _install exits on version mismatch; swallow to protect the host process.
+        pass
+    except Exception as e:
+        sys.stderr.write(f"[clearml-truen-patch] auto-apply skipped: {e}\n")
+
+
+_safe_apply()

clearml_truen_patch-2.1.3.post1/clearml_truen_patch/_install.py

@@ -0,0 +1,139 @@
+"""Overlay clearml_truen_patch content onto the installed clearml package.
+
+Run automatically by setup.py after install, and can also be invoked manually:
+    python -m clearml_truen_patch._install
+    python -m clearml_truen_patch._install --restore
+"""
+import hashlib
+import json
+import shutil
+import sys
+from pathlib import Path
+
+REQUIRED_CLEARML_VERSION = "2.1.3"
+MARKER_FILENAME = ".truen_patch_applied.json"
+BACKUP_DIRNAME = "clearml._truen_backup"
+
+
+def _patch_version():
+    try:
+        from importlib.metadata import version
+        return version("clearml-truen-patch")
+    except Exception:
+        return "unknown"
+
+
+def _source_fingerprint(src: Path) -> str:
+    """Hash of (relative path + content) for every source file; used to detect
+    when the installed overlay source has changed between interpreter runs."""
+    h = hashlib.sha256()
+    for f in sorted(src.rglob("*")):
+        if not f.is_file():
+            continue
+        rel = str(f.relative_to(src)).replace("\\", "/")
+        h.update(rel.encode("utf-8"))
+        h.update(b"\0")
+        h.update(f.read_bytes())
+        h.update(b"\0")
+    return h.hexdigest()
+
+
+def _find_clearml_dir():
+    try:
+        import clearml
+    except ImportError:
+        sys.exit(
+            "[clearml-truen-patch] 'clearml' is not installed. "
+            f"Install clearml=={REQUIRED_CLEARML_VERSION} first."
+        )
+
+    version = getattr(clearml, "__version__", None)
+    if version != REQUIRED_CLEARML_VERSION:
+        sys.exit(
+            f"[clearml-truen-patch] clearml=={REQUIRED_CLEARML_VERSION} required, "
+            f"found {version}. Aborting to avoid corrupting an unsupported version."
+        )
+    return Path(clearml.__file__).resolve().parent
+
+
+def _source_dir():
+    return Path(__file__).resolve().parent
+
+
+def needs_apply(dst: Path, src: Path) -> bool:
+    marker = dst / MARKER_FILENAME
+    if not marker.exists():
+        return True
+    try:
+        data = json.loads(marker.read_text(encoding="utf-8"))
+    except Exception:
+        return True
+    if data.get("clearml_version") != REQUIRED_CLEARML_VERSION:
+        return True
+    if data.get("patch_version") != _patch_version():
+        return True
+    if data.get("source_fingerprint") != _source_fingerprint(src):
+        return True
+    return False
+
+
+def run(force: bool = True):
+    dst = _find_clearml_dir()
+    src = _source_dir()
+
+    if not force and not needs_apply(dst, src):
+        return
+
+    backup = dst.parent / BACKUP_DIRNAME
+    if not backup.exists():
+        print(f"[clearml-truen-patch] Backing up original clearml -> {backup}")
+        shutil.copytree(dst, backup)
+
+    copied = []
+    skip_names = {"_install.py", "_autoapply.py", "__pycache__", MARKER_FILENAME}
+    for f in src.rglob("*"):
+        if any(part in skip_names for part in f.relative_to(src).parts):
+            continue
+        if not f.is_file():
+            continue
+        rel = f.relative_to(src)
+        target = dst / rel
+        target.parent.mkdir(parents=True, exist_ok=True)
+        shutil.copy2(f, target)
+        copied.append(str(rel).replace("\\", "/"))
+
+    marker = dst / MARKER_FILENAME
+    marker.write_text(
+        json.dumps(
+            {
+                "clearml_version": REQUIRED_CLEARML_VERSION,
+                "patch_version": _patch_version(),
+                "source_fingerprint": _source_fingerprint(src),
+                "files": copied,
+            },
+            indent=2,
+        ),
+        encoding="utf-8",
+    )
+    print(f"[clearml-truen-patch] Overlay applied: {len(copied)} files -> {dst}")
+
+
+def restore():
+    try:
+        import clearml
+    except ImportError:
+        sys.exit("[clearml-truen-patch] clearml is not installed; nothing to restore.")
+    dst = Path(clearml.__file__).resolve().parent
+    backup = dst.parent / BACKUP_DIRNAME
+    if not backup.exists():
+        sys.exit(f"[clearml-truen-patch] No backup found at {backup}")
+    shutil.rmtree(dst)
+    shutil.copytree(backup, dst)
+    print(f"[clearml-truen-patch] Restored original clearml from {backup}")
+
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1 and sys.argv[1] == "--restore":
+        restore()
+    else:
+        run(force=True)

clearml_truen_patch-2.1.3.post1/clearml_truen_patch/automation/__init__.py

@@ -0,0 +1,29 @@
+from .job import ClearmlJob
+from .controller import PipelineController, PipelineDecorator
+from .optimization import GridSearch, RandomSearch, HyperParameterOptimizer, Objective
+from .parameters import (
+    UniformParameterRange,
+    DiscreteParameterRange,
+    UniformIntegerParameterRange,
+    ParameterSet,
+    LogUniformParameterRange,
+)
+from .scheduler import TaskScheduler
+from .trigger import TriggerScheduler
+
+__all__ = [
+    "UniformParameterRange",
+    "DiscreteParameterRange",
+    "UniformIntegerParameterRange",
+    "ParameterSet",
+    "LogUniformParameterRange",
+    "GridSearch",
+    "RandomSearch",
+    "HyperParameterOptimizer",
+    "Objective",
+    "ClearmlJob",
+    "PipelineController",
+    "TaskScheduler",
+    "TriggerScheduler",
+    "PipelineDecorator",
+]