metis-optimizer 0.0.2__py3-none-any.whl

metis_optimizer-0.0.2.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,5 @@
+metis_optimizer_setup/__init__.py,sha256=u1hrt13xvmSD25Zx5-t6XseJEvzmjme9LLYIc0NkVZM,1626
+metis_optimizer_setup/__main__.py,sha256=XnpTzfJc0T-j9tHtdkA8ovTr1c139ffTewcJGhxYDaM,49
+metis_optimizer-0.0.2.dist-info/METADATA,sha256=Z7yqhHtUJSiXVvDZqBNnOCA0PNp3eARVjAI9Fdl6Swk,1105
+metis_optimizer-0.0.2.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+metis_optimizer-0.0.2.dist-info/RECORD,,

metis_optimizer-0.0.2.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.32.4
+Root-Is-Purelib: true
+Tag: py3-none-any

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_setup/__main__.py

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