metis-optimizer 0.0.2__tar.gz

metis_optimizer-0.0.2/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.5
+Name: metis-optimizer
+Version: 0.0.2
+Summary: MetisPrism setup helper: shows how to install MetisPrism tools from your account and checks your local Python and GPU setup.
+Project-URL: Homepage, https://metisprism.ai
+Author: MetisAI Labs
+License-Expression: Apache-2.0
+Keywords: gpu,metisprism,ml-infrastructure
+Classifier: Development Status :: 2 - Pre-Alpha
+Classifier: Environment :: Console
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+
+# metis-optimizer
+
+MetisPrism setup helper for ML teams running training, fine-tuning and serving on MetisPrism.
+
+```
+pip install metis-optimizer
+python -m metis_optimizer_setup
+```
+
+It prints how to install the MetisPrism CLI and SDKs from your MetisPrism account, and checks your local setup: Python
+version, platform, NVIDIA or AMD GPU tools on PATH, and an installed PyTorch. It reads only your local environment and
+sends nothing over the network.
+
+The full MetisPrism SDK will be published at public launch.

metis_optimizer-0.0.2/README.md

@@ -0,0 +1,14 @@
+# metis-optimizer
+
+MetisPrism setup helper for ML teams running training, fine-tuning and serving on MetisPrism.
+
+```
+pip install metis-optimizer
+python -m metis_optimizer_setup
+```
+
+It prints how to install the MetisPrism CLI and SDKs from your MetisPrism account, and checks your local setup: Python
+version, platform, NVIDIA or AMD GPU tools on PATH, and an installed PyTorch. It reads only your local environment and
+sends nothing over the network.
+
+The full MetisPrism SDK will be published at public launch.

metis_optimizer-0.0.2/pyproject.toml

@@ -0,0 +1,20 @@
+[build-system]
+requires = ["hatchling>=1.25"]
+build-backend = "hatchling.build"
+
+[project]
+name = "metis-optimizer"
+version = "0.0.2"
+description = "MetisPrism setup helper: shows how to install MetisPrism tools from your account and checks your local Python and GPU setup."
+readme = "README.md"
+license = "Apache-2.0"
+requires-python = ">=3.9"
+authors = [{ name = "MetisAI Labs" }]
+keywords = ["metisprism", "gpu", "ml-infrastructure"]
+classifiers = ["Development Status :: 2 - Pre-Alpha", "License :: OSI Approved :: Apache Software License", "Programming Language :: Python :: 3", "Environment :: Console"]
+
+[project.urls]
+Homepage = "https://metisprism.ai"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/metis_optimizer_setup"]

metis_optimizer-0.0.2/src/metis_optimizer_setup/__init__.py

@@ -0,0 +1,47 @@
+"""MetisPrism setup helper: install guidance and a local environment check. Sends nothing over the network."""
+
+import importlib.util
+import platform
+import shutil
+import sys
+
+__all__ = ["doctor", "main"]
+__version__ = "0.0.2"
+
+INSTALL = (
+    "The MetisPrism CLI and SDKs are installed from your MetisPrism account:\n"
+    "  Studio -> Install the CLI (signed download), or ask your MetisPrism contact for access.\n"
+    "Docs: https://metisprism.ai"
+)
+
+
+def doctor() -> dict:
+    """Return a summary of the local environment relevant to running ML workloads with MetisPrism."""
+    report = {
+        "python": platform.python_version(),
+        "platform": f"{platform.system()} {platform.machine()}",
+        "nvidia_smi": shutil.which("nvidia-smi") is not None,
+        "rocm_smi": shutil.which("rocm-smi") is not None or shutil.which("amd-smi") is not None,
+        "torch": None,
+        "accelerators": 0,
+    }
+    if importlib.util.find_spec("torch") is not None:
+        try:
+            import torch  # noqa: PLC0415
+
+            report["torch"] = torch.__version__
+            if torch.cuda.is_available():
+                report["accelerators"] = torch.cuda.device_count()
+        except Exception as exc:  # a broken torch install is reported, not raised
+            report["torch"] = f"present but failed to import: {exc.__class__.__name__}"
+    return report
+
+
+def main() -> int:
+    print(INSTALL)
+    print()
+    for key, value in doctor().items():
+        print(f"  {key:13} {value}")
+    if sys.version_info < (3, 9):
+        print("\nPython 3.9 or newer is required by the MetisPrism SDK.")
+    return 0

metis_optimizer-0.0.2/src/metis_optimizer_setup/__main__.py

@@ -0,0 +1,5 @@
+import sys
+
+from . import main
+
+sys.exit(main())