gsml 1.0.0__tar.gz

gsml-1.0.0/PKG-INFO

@@ -0,0 +1,49 @@
+Metadata-Version: 2.4
+Name: gsml
+Version: 1.0.0
+Summary: General Scientific ML toolkit
+Author-email: Dennis Liu <dennisl@udel.edu>
+License: MIT
+Project-URL: Homepage, https://github.com/ByteDennis/gsml
+Project-URL: Repository, https://github.com/ByteDennis/gsml
+Project-URL: Issues, https://github.com/ByteDennis/gsml/issues
+Keywords: mlops,sqlite
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: springs>=1.13.0
+Requires-Dist: omegaconf>=2.3.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: tqdm>=4.60.0
+Requires-Dist: loguru>=0.7.0
+Requires-Dist: numpy>=1.24.0
+Provides-Extra: torch
+Requires-Dist: torch>=2.0.0; extra == "torch"
+Provides-Extra: sklearn
+Requires-Dist: scikit-learn>=1.0.0; extra == "sklearn"
+Requires-Dist: joblib>=1.0.0; extra == "sklearn"
+Provides-Extra: tune
+Requires-Dist: optuna>=3.0.0; extra == "tune"
+Requires-Dist: pandas>=2.0.0; extra == "tune"
+Requires-Dist: requests>=2.28.0; extra == "tune"
+Provides-Extra: bench
+Requires-Dist: pandas>=2.0.0; extra == "bench"
+Provides-Extra: wandb
+Requires-Dist: wandb>=0.15.0; extra == "wandb"
+Provides-Extra: all
+Requires-Dist: torch>=2.0.0; extra == "all"
+Requires-Dist: scikit-learn>=1.0.0; extra == "all"
+Requires-Dist: joblib>=1.0.0; extra == "all"
+Requires-Dist: optuna>=3.0.0; extra == "all"
+Requires-Dist: pandas>=2.0.0; extra == "all"
+Requires-Dist: requests>=2.28.0; extra == "all"
+Requires-Dist: wandb>=0.15.0; extra == "all"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: build>=1.0.0; extra == "dev"

gsml-1.0.0/gsml/__init__.py

@@ -0,0 +1,35 @@
+# ruff: noqa: F403 F405
+"""
+gsml — lightweight experiment tracking and ML infrastructure.
+
+Public API re-exports all submodule contents for convenient access:
+    import gsml as lib
+    lib.Tracker(...)
+    lib.METRIC_REGISTRY[...]
+"""
+
+from .track import Track, tracked
+
+# Re-export all submodule contents for `import gsml as lib` usage
+from .config import *
+from .db import *
+from .dispatch import *
+from .log import *
+from .util import *
+from .plan import *
+from .enums import *
+from .dtypes import *
+from .tracker import *
+from .metric import *
+from .prompt import *
+from .data import *
+from .testing import *
+
+# bench and tune have heavy deps (pandas, numpy, optuna) — lazy import submodules
+# but eagerly re-export their contents for lib.X access
+from .bench import *
+from .tune import *
+
+__all__ = ["Track", "tracked"]
+
+__version__ = "1.0.0"

gsml-1.0.0/gsml/_compat.py

@@ -0,0 +1,118 @@
+"""Optional dependency guards for gsml.
+
+Provides lazy imports that give clear error messages when optional
+packages are not installed, instead of crashing at import time.
+
+Usage:
+    from gsml._compat import torch, require
+    require("torch")  # raises ImportError with install instructions if missing
+    torch.cuda.is_available()  # works normally if installed
+"""
+
+import importlib
+from types import ModuleType
+
+
+class _MissingModule:
+    """Proxy object that raises ImportError on any attribute access."""
+
+    def __init__(self, name: str, package: str, extra: str):
+        self._name = name
+        self._package = package
+        self._extra = extra
+
+    def _raise(self):
+        raise ImportError(
+            f"{self._package} is required but not installed. "
+            f"Install it with: pip install gsml[{self._extra}]"
+        )
+
+    def __getattr__(self, item):
+        if item.startswith("_"):
+            return object.__getattribute__(self, item)
+        self._raise()
+
+    def __call__(self, *args, **kwargs):
+        self._raise()
+
+    def __bool__(self):
+        return False
+
+    def __repr__(self):
+        return f"<MissingModule: {self._name} (pip install gsml[{self._extra}])>"
+
+
+def _lazy_import(module_name: str, package: str, extra: str) -> ModuleType:
+    """Import a module, returning a proxy if not installed."""
+    try:
+        return importlib.import_module(module_name)
+    except ImportError:
+        return _MissingModule(module_name, package, extra)
+
+
+def has_torch() -> bool:
+    """Check if PyTorch is available."""
+    try:
+        import torch  # noqa: F811
+        return True
+    except ImportError:
+        return False
+
+
+def has_sklearn() -> bool:
+    """Check if scikit-learn is available."""
+    try:
+        import sklearn  # noqa: F811
+        return True
+    except ImportError:
+        return False
+
+
+def has_optuna() -> bool:
+    """Check if Optuna is available."""
+    try:
+        import optuna  # noqa: F811
+        return True
+    except ImportError:
+        return False
+
+
+def has_pandas() -> bool:
+    """Check if pandas is available."""
+    try:
+        import pandas  # noqa: F811
+        return True
+    except ImportError:
+        return False
+
+
+def require(module_name: str):
+    """Require a module, raising ImportError with instructions if missing."""
+    extras = {
+        "torch": "torch",
+        "sklearn": "sklearn",
+        "scikit-learn": "sklearn",
+        "optuna": "tune",
+        "pandas": "bench",
+        "joblib": "sklearn",
+        "wandb": "wandb",
+        "requests": "tune",
+    }
+    extra = extras.get(module_name, "all")
+    package = module_name
+    try:
+        importlib.import_module(module_name)
+    except ImportError:
+        raise ImportError(
+            f"{package} is required but not installed. "
+            f"Install it with: pip install gsml[{extra}]"
+        )
+
+
+# Lazy singletons — safe to import at module level
+torch = _lazy_import("torch", "torch", "torch")
+sklearn = _lazy_import("sklearn", "scikit-learn", "sklearn")
+optuna = _lazy_import("optuna", "optuna", "tune")
+pandas = _lazy_import("pandas", "pandas", "bench")
+joblib = _lazy_import("joblib", "joblib", "sklearn")
+wandb = _lazy_import("wandb", "wandb", "wandb")