mlenvdoctor 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

mlenvdoctor/__init__.py

@@ -1,4 +1,3 @@
 """ML Environment Doctor - Diagnose and fix ML environments for LLM fine-tuning."""
 
 __version__ = "0.1.0"
-

mlenvdoctor/cli.py

@@ -8,7 +8,7 @@ from . import __version__
 from .diagnose import diagnose_env, print_diagnostic_table
 from .dockerize import generate_dockerfile, generate_service_template
 from .fix import auto_fix
-from .gpu import benchmark_gpu_ops, smoke_test_lora, test_model
+from .gpu import benchmark_gpu_ops, smoke_test_lora, test_model as gpu_test_model
 from .utils import console
 
 app = typer.Typer(
@@ -21,14 +21,21 @@ app = typer.Typer(
 def version_callback(value: bool):
     """Print version and exit."""
     if value:
-        console.print(f"[bold blue]ML Environment Doctor[/bold blue] version [cyan]{__version__}[/cyan]")
+        console.print(
+            f"[bold blue]ML Environment Doctor[/bold blue] version [cyan]{__version__}[/cyan]"
+        )
         raise typer.Exit()
 
 
 @app.callback()
 def main(
     version: Optional[bool] = typer.Option(
-        None, "--version", "-v", callback=version_callback, is_eager=True, help="Show version and exit"
+        None,
+        "--version",
+        "-v",
+        callback=version_callback,
+        is_eager=True,
+        help="Show version and exit",
     ),
 ):
     """ML Environment Doctor - Diagnose & fix ML environments for LLM fine-tuning."""
@@ -37,7 +44,9 @@ def main(
 
 @app.command()
 def diagnose(
-    full: bool = typer.Option(False, "--full", "-f", help="Run full diagnostics including GPU benchmarks"),
+    full: bool = typer.Option(
+        False, "--full", "-f", help="Run full diagnostics including GPU benchmarks"
+    ),
 ):
     """
     🔍 Diagnose your ML environment.
@@ -85,8 +94,12 @@ def fix(
 @app.command()
 def dockerize(
     model: Optional[str] = typer.Argument(None, help="Model name (mistral-7b, tinyllama, gpt2)"),
-    service: bool = typer.Option(False, "--service", "-s", help="Generate FastAPI service template"),
-    output: str = typer.Option("Dockerfile.mlenvdoctor", "--output", "-o", help="Output Dockerfile name"),
+    service: bool = typer.Option(
+        False, "--service", "-s", help="Generate FastAPI service template"
+    ),
+    output: str = typer.Option(
+        "Dockerfile.mlenvdoctor", "--output", "-o", help="Output Dockerfile name"
+    ),
 ):
     """
     🐳 Generate Dockerfile for ML fine-tuning.
@@ -105,8 +118,8 @@ def dockerize(
     console.print("[bold green]✅ Dockerfile generated![/bold green]")
 
 
-@app.command()
-def test_model(
+@app.command(name="test-model")
+def test_model_cmd(
     model: str = typer.Argument("tinyllama", help="Model to test (tinyllama, gpt2, mistral-7b)"),
 ):
     """
@@ -115,7 +128,7 @@ def test_model(
     Tests model loading and forward pass to verify fine-tuning readiness.
     """
     console.print(f"[bold blue]🧪 Testing model: {model}[/bold blue]\n")
-    success = test_model(model_name=model)
+    success = gpu_test_model(model_name=model)
     if success:
         console.print()
         console.print("[bold green]✅ Model test passed! Ready for fine-tuning.[/bold green]")
@@ -139,7 +152,9 @@ def smoke_test():
         console.print("[bold green]✅ Smoke test passed! Environment is ready.[/bold green]")
     else:
         console.print()
-        console.print("[bold red]❌ Smoke test failed. Run 'mlenvdoctor diagnose' for details.[/bold red]")
+        console.print(
+            "[bold red]❌ Smoke test failed. Run 'mlenvdoctor diagnose' for details.[/bold red]"
+        )
         raise typer.Exit(1)
 
 
@@ -150,4 +165,3 @@ def main_cli():
 
 if __name__ == "__main__":
     main_cli()
-

mlenvdoctor/diagnose.py

@@ -2,8 +2,7 @@
 
 import importlib
 import re
-import subprocess
-from typing import Dict, List, Optional, Tuple
+from typing import List, Optional, Tuple
 
 try:
     import torch
@@ -12,16 +11,7 @@ except ImportError:
 
 from rich.table import Table
 
-from .utils import (
-    check_command_exists,
-    console,
-    format_size,
-    get_home_config_dir,
-    print_error,
-    print_info,
-    print_warning,
-    run_command,
-)
+from .utils import check_command_exists, console, format_size, get_home_config_dir, run_command
 
 
 class DiagnosticIssue:
@@ -164,7 +154,7 @@ def check_pytorch_cuda() -> List[DiagnosticIssue]:
                         name="PyTorch Version",
                         status="WARN - Old version",
                         severity="warning",
-                        fix=f"Upgrade: pip install torch>=2.4.0 --index-url https://download.pytorch.org/whl/cu124",
+                        fix="Upgrade: pip install torch>=2.4.0 --index-url https://download.pytorch.org/whl/cu124",
                         details=f"Current: {torch_version}, Recommended: >=2.4.0",
                     )
                 )
@@ -237,7 +227,7 @@ def check_ml_libraries() -> List[DiagnosticIssue]:
                     issues.append(
                         DiagnosticIssue(
                             name=f"{lib_name}",
-                            status=f"PASS - Installed",
+                            status="PASS - Installed",
                             severity="info",
                             fix="",
                             details=f"Version: {version}",
@@ -275,7 +265,6 @@ def check_gpu_memory() -> List[DiagnosticIssue]:
         free_mem, total_mem = torch.cuda.mem_get_info(0)
         free_gb = free_mem / (1024**3)
         total_gb = total_mem / (1024**3)
-        used_gb = total_gb - free_gb
 
         if free_gb < 8:
             issues.append(
@@ -377,7 +366,18 @@ def check_docker_gpu() -> List[DiagnosticIssue]:
         return issues
 
     try:
-        result = run_command(["docker", "run", "--rm", "--gpus", "all", "nvidia/cuda:12.4.0-base-ubuntu22.04", "nvidia-smi"], timeout=30)
+        result = run_command(
+            [
+                "docker",
+                "run",
+                "--rm",
+                "--gpus",
+                "all",
+                "nvidia/cuda:12.4.0-base-ubuntu22.04",
+                "nvidia-smi",
+            ],
+            timeout=30,
+        )
         if result.returncode == 0:
             issues.append(
                 DiagnosticIssue(
@@ -460,7 +460,11 @@ def diagnose_env(full: bool = False) -> List[DiagnosticIssue]:
 
 def print_diagnostic_table(issues: List[DiagnosticIssue]) -> None:
     """Print diagnostic results as a Rich table."""
-    table = Table(title="ML Environment Doctor - Diagnostic Results", show_header=True, header_style="bold magenta")
+    table = Table(
+        title="ML Environment Doctor - Diagnostic Results",
+        show_header=True,
+        header_style="bold magenta",
+    )
     table.add_column("Issue", style="cyan", no_wrap=False)
     table.add_column("Status", style="bold")
     table.add_column("Severity", style="yellow")
@@ -474,7 +478,9 @@ def print_diagnostic_table(issues: List[DiagnosticIssue]) -> None:
 
     # Summary
     critical_count = sum(1 for i in issues if i.severity == "critical" and "FAIL" in i.status)
-    warning_count = sum(1 for i in issues if i.severity == "warning" and ("WARN" in i.status or "FAIL" in i.status))
+    warning_count = sum(
+        1 for i in issues if i.severity == "warning" and ("WARN" in i.status or "FAIL" in i.status)
+    )
     pass_count = sum(1 for i in issues if "PASS" in i.status)
 
     console.print()
@@ -485,9 +491,12 @@ def print_diagnostic_table(issues: List[DiagnosticIssue]) -> None:
         console.print(f"[red]❌ Critical Issues: {critical_count}[/red]")
 
     if critical_count == 0 and warning_count == 0:
-        console.print("\n[bold green]🎉 Your ML environment looks ready for fine-tuning![/bold green]")
+        console.print(
+            "\n[bold green]🎉 Your ML environment looks ready for fine-tuning![/bold green]"
+        )
     elif critical_count > 0:
         console.print("\n[bold red]⚠️  Please fix critical issues before proceeding.[/bold red]")
     else:
-        console.print("\n[bold yellow]💡 Consider addressing warnings for optimal performance.[/bold yellow]")
-
+        console.print(
+            "\n[bold yellow]💡 Consider addressing warnings for optimal performance.[/bold yellow]"
+        )

mlenvdoctor/dockerize.py

@@ -3,9 +3,7 @@
 from pathlib import Path
 from typing import Optional
 
-from rich.console import Console
-
-from .utils import console, print_error, print_info, print_success
+from .utils import console, print_info, print_success
 
 # Model-specific templates
 MODEL_TEMPLATES = {
@@ -149,9 +147,9 @@ CMD ["python", "train.py"]
     console.print("[bold]Build and run:[/bold]")
     console.print(f"[cyan]  docker build -f {output_file} -t mlenvdoctor .[/cyan]")
     if service:
-        console.print(f"[cyan]  docker run --gpus all -p 8000:8000 mlenvdoctor[/cyan]")
+        console.print("[cyan]  docker run --gpus all -p 8000:8000 mlenvdoctor[/cyan]")
     else:
-        console.print(f"[cyan]  docker run --gpus all -v $(pwd)/data:/app/data mlenvdoctor[/cyan]")
+        console.print("[cyan]  docker run --gpus all -v $(pwd)/data:/app/data mlenvdoctor[/cyan]")
 
     return output_path
 
@@ -201,4 +199,3 @@ if __name__ == "__main__":
     output_path.write_text(service_content, encoding="utf-8")
     print_success(f"Generated service template: {output_file}")
     return output_path
-

mlenvdoctor/fix.py

@@ -1,14 +1,10 @@
 """Auto-fix and requirements generation for ML Environment Doctor."""
 
-import subprocess
 import sys
 from pathlib import Path
-from typing import List, Optional
+from typing import Optional
 
-from rich.console import Console
-from rich.progress import Progress, SpinnerColumn, TextColumn
-
-from .diagnose import DiagnosticIssue, diagnose_env
+from .diagnose import diagnose_env
 from .utils import (
     check_command_exists,
     console,
@@ -39,7 +35,9 @@ ML_STACKS = {
 }
 
 
-def generate_requirements_txt(stack: str = "trl-peft", output_file: str = "requirements-mlenvdoctor.txt") -> Path:
+def generate_requirements_txt(
+    stack: str = "trl-peft", output_file: str = "requirements-mlenvdoctor.txt"
+) -> Path:
     """Generate requirements.txt file."""
     if stack not in ML_STACKS:
         print_error(f"Unknown stack: {stack}. Available: {list(ML_STACKS.keys())}")
@@ -61,7 +59,9 @@ def generate_requirements_txt(stack: str = "trl-peft", output_file: str = "requi
             content = "# Standard PyTorch (CPU or CUDA)\n\n"
     except ImportError:
         content = "# PyTorch installation\n"
-        content += "# For CUDA: pip install torch --index-url https://download.pytorch.org/whl/cu124\n"
+        content += (
+            "# For CUDA: pip install torch --index-url https://download.pytorch.org/whl/cu124\n"
+        )
         content += "# For CPU: pip install torch\n\n"
 
     content += "\n".join(requirements)
@@ -72,7 +72,9 @@ def generate_requirements_txt(stack: str = "trl-peft", output_file: str = "requi
     return output_path
 
 
-def generate_conda_env(stack: str = "trl-peft", output_file: str = "environment-mlenvdoctor.yml") -> Path:
+def generate_conda_env(
+    stack: str = "trl-peft", output_file: str = "environment-mlenvdoctor.yml"
+) -> Path:
     """Generate conda environment file."""
     if stack not in ML_STACKS:
         print_error(f"Unknown stack: {stack}. Available: {list(ML_STACKS.keys())}")
@@ -197,7 +199,11 @@ def create_virtualenv(env_name: str = ".venv") -> Optional[Path]:
 
         venv.create(env_path, with_pip=True)
         print_success(f"Virtual environment created: {env_name}")
-        print_info(f"Activate with: {'.venv\\Scripts\\activate' if sys.platform == 'win32' else 'source .venv/bin/activate'}")
+        if sys.platform == "win32":
+            activate_cmd = r".venv\Scripts\activate"
+        else:
+            activate_cmd = "source .venv/bin/activate"
+        print_info(f"Activate with: {activate_cmd}")
         return env_path
     except Exception as e:
         print_error(f"Failed to create virtual environment: {e}")
@@ -243,7 +249,6 @@ def auto_fix(use_conda: bool = False, create_venv: bool = False, stack: str = "t
         if install.lower() in ["y", "yes"]:
             return install_requirements(str(req_file), use_conda=use_conda)
         else:
-            print_info(f"Requirements file generated. Install manually with:")
+            print_info("Requirements file generated. Install manually with:")
             console.print(f"[cyan]  pip install -r {req_file}[/cyan]")
             return True
-

mlenvdoctor/gpu.py

@@ -1,7 +1,7 @@
 """GPU benchmarks and smoke tests for ML Environment Doctor."""
 
 import time
-from typing import Dict, List, Optional
+from typing import Dict
 
 try:
     import torch
@@ -9,7 +9,6 @@ except ImportError:
     torch = None  # type: ignore
 
 from rich.console import Console
-from rich.progress import Progress, SpinnerColumn, TextColumn
 
 from .utils import print_error, print_info, print_success
 
@@ -78,7 +77,8 @@ def smoke_test_lora() -> bool:
                 if tokenizer.pad_token is None:
                     tokenizer.pad_token = tokenizer.eos_token
                 model = AutoModelForCausalLM.from_pretrained(
-                    model_name, torch_dtype=torch.float16 if device.type == "cuda" else torch.float32
+                    model_name,
+                    torch_dtype=torch.float16 if device.type == "cuda" else torch.float32,
                 ).to(device)
 
             # Configure LoRA
@@ -101,8 +101,7 @@ def smoke_test_lora() -> bool:
             # Forward pass
             with console.status("[bold green]Running forward pass..."):
                 with torch.no_grad():
-                    outputs = model(**inputs)
-                    loss = outputs.loss if hasattr(outputs, "loss") else None
+                    _ = model(**inputs)
 
             print_success("LoRA smoke test passed!")
             return True
@@ -146,7 +145,9 @@ def test_model(model_name: str = "tinyllama") -> bool:
         # Estimate memory requirements (rough)
         if "7b" in actual_model_name.lower() or "7B" in actual_model_name:
             if free_gb < 16:
-                print_error(f"Insufficient GPU memory: {free_gb:.1f}GB free, need ~16GB for 7B model")
+                print_error(
+                    f"Insufficient GPU memory: {free_gb:.1f}GB free, need ~16GB for 7B model"
+                )
                 return False
 
         with console.status(f"[bold green]Loading {actual_model_name}..."):
@@ -166,7 +167,7 @@ def test_model(model_name: str = "tinyllama") -> bool:
                 inputs = tokenizer(dummy_text, return_tensors="pt").to(device)
 
                 with torch.no_grad():
-                    outputs = model(**inputs)
+                    _ = model(**inputs)
 
                 print_success(f"Model {actual_model_name} loaded and tested successfully!")
                 return True
@@ -181,4 +182,3 @@ def test_model(model_name: str = "tinyllama") -> bool:
     except Exception as e:
         print_error(f"Model test error: {e}")
         return False
-

mlenvdoctor/utils.py

@@ -7,7 +7,6 @@ from typing import List, Optional, Tuple
 
 from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn
-from rich.text import Text
 
 console = Console()
 
@@ -103,5 +102,3 @@ def format_size(size_bytes: int) -> str:
 def get_python_version() -> Tuple[int, int, int]:
     """Get Python version as tuple."""
     return sys.version_info[:3]
-
-

{mlenvdoctor-0.1.0.dist-info → mlenvdoctor-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlenvdoctor
-Version: 0.1.0
+Version: 0.1.1
 Summary: Diagnose & fix ML environments for LLM fine-tuning
 Author: ML Environment Doctor Contributors
 License: MIT
@@ -34,7 +34,7 @@ Description-Content-Type: text/markdown
 
 [![Python 3.8+](https://img.shields.io/badge/python-3.8+-blue.svg)](https://www.python.org/downloads/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![PyPI version](https://badge.fury.io/py/mlenvdoctor.svg)](https://badge.fury.io/py/mlenvdoctor)
+[![PyPI](https://img.shields.io/pypi/v/mlenvdoctor.svg)](https://pypi.org/project/mlenvdoctor/)
 
 > **Single command fixes 90% of "my torch.cuda.is_available() is False" issues.**
 

mlenvdoctor-0.1.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+mlenvdoctor/__init__.py,sha256=vYK9Wp5kAcHKL7njV76xMfPA24ILdxJvIhoumF6-Sz4,110
+mlenvdoctor/cli.py,sha256=aQ2rpjxsfMDwYAVKsn7cHSar42AZJTrr1tSTl-iM0L4,5488
+mlenvdoctor/diagnose.py,sha256=xa3aqCornGApMJkEWQNIGHwNBRhGA3ud1hBQ6wIVhVQ,17099
+mlenvdoctor/dockerize.py,sha256=AC8HX5sRkSFAM0O0caBnKW4HAdS49MVmMcsplKEDXI4,5562
+mlenvdoctor/fix.py,sha256=P4Qce41LLgjaHugbMFFSg7ldfsSSNDFBz5_T_YA9mig,8945
+mlenvdoctor/gpu.py,sha256=iuiLAW8lZLBpuUL1yapOkx5VYLtY_i1SwK9cE5koZTE,6129
+mlenvdoctor/utils.py,sha256=2gtbiJogEI33IpOLHGEfks6b7Jd1Y7pyfojW9wpYsjU,2893
+mlenvdoctor-0.1.1.dist-info/METADATA,sha256=NQHbKeu7KnZfHWPJcqQEYUXZqIRL7ORCesdMyyqFKU8,8887
+mlenvdoctor-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mlenvdoctor-0.1.1.dist-info/entry_points.txt,sha256=Y-WH-ANeiTdECIaqi_EB3ZEf_kACkvsYBHnNhXsCI4k,52
+mlenvdoctor-0.1.1.dist-info/licenses/LICENSE,sha256=rGHdyWGvGWYnEFlthqtB-RtRCTa7WaAOElom5qD-nHw,1114
+mlenvdoctor-0.1.1.dist-info/RECORD,,

mlenvdoctor-0.1.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-mlenvdoctor/__init__.py,sha256=hPyn5R1E9SDxyqhMb9AUsDw-TRxj9K2AWcadl5LmvqY,112
-mlenvdoctor/cli.py,sha256=DjF6UYC4QlrGSvbmG3wokX0urAWls9hu0FSIh-jZyYA,5305
-mlenvdoctor/diagnose.py,sha256=zk_E3UF2OlJr6ZFx3OjejyswgcZWX7qz1BHDXCB-vmk,16980
-mlenvdoctor/dockerize.py,sha256=q7afAUSkpL3RHSmbMZy2G9VZ7yRbus4xDADbSjIIuJM,5615
-mlenvdoctor/fix.py,sha256=V-mK30r2xSk-_3uuEHqH01iA9Vt8mjjlCisb4kL_A_Q,8959
-mlenvdoctor/gpu.py,sha256=PTS_dj6JaAoXKHQNzqgh8xOuZHomPoQxuLQMGWQXHqQ,6239
-mlenvdoctor/utils.py,sha256=fAZHgVX6iyOpcd4NU-oSyLjFMO1AXnfgOL5_KL_-0Po,2925
-mlenvdoctor-0.1.0.dist-info/METADATA,sha256=mZnlvzRuaaW9ydTZf5HsP4-wtDs7LUKUEde2xu_STmo,8889
-mlenvdoctor-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mlenvdoctor-0.1.0.dist-info/entry_points.txt,sha256=Y-WH-ANeiTdECIaqi_EB3ZEf_kACkvsYBHnNhXsCI4k,52
-mlenvdoctor-0.1.0.dist-info/licenses/LICENSE,sha256=rGHdyWGvGWYnEFlthqtB-RtRCTa7WaAOElom5qD-nHw,1114
-mlenvdoctor-0.1.0.dist-info/RECORD,,