dayhoff-tools 1.13.16__tar.gz → 1.13.18__tar.gz

{dayhoff_tools-1.13.16 → dayhoff_tools-1.13.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dayhoff-tools
-Version: 1.13.16
+Version: 1.13.18
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com

{dayhoff_tools-1.13.16 → dayhoff_tools-1.13.18}/dayhoff_tools/cli/engines_studios/engine-studio-cli.md

@@ -88,6 +88,7 @@ dh engine2 launch e2e-engine --type cpu --user testuser1
 ✓ EC2 instance launched: i-1234567890abcdef0
 
 ⏳ Waiting for engine to be ready (typically 2-3 minutes)...
+# Note: GPU engines show "5-7 minutes" due to driver installation + reboot
 
 Progress ████████████░░░░░░░░░░ 60%
   [5s] Instance Running

{dayhoff_tools-1.13.16 → dayhoff_tools-1.13.18}/dayhoff_tools/cli/engines_studios/engine_commands.py

@@ -190,7 +190,10 @@ def launch_engine(
             return
 
         # Wait for readiness with progress updates
-        click.echo(f"\n⏳ Waiting for engine to be ready (typically 2-3 minutes)...\n")
+        # GPU engines take longer due to driver installation and reboot
+        is_gpu = engine_type.lower() not in ("cpu", "cpumax")
+        wait_time = "6-8 minutes" if is_gpu else "2-3 minutes"
+        click.echo(f"\n⏳ Waiting for engine to be ready (typically {wait_time})...\n")
 
         try:
             _final_status = wait_with_progress(
@@ -1121,19 +1124,28 @@ def debug_engine(name_or_id: str, env: Optional[str]):
         click.echo(f"Current Stage: {readiness.get('current_stage', 'unknown')}")
         click.echo(f"\nBootstrap Stages:")
 
-        stages = readiness.get("stages", [])
-        for i, stage in enumerate(stages, 1):
-            status = stage.get("status", "unknown")
+        # Deduplicate stages by name, keeping the latest status for each
+        # (bootstrap reports both "starting" and "completed" for each stage)
+        stages_by_name: dict[str, dict] = {}
+        for stage in readiness.get("stages", []):
             name = stage.get("name", "unknown")
+            stages_by_name[name] = stage  # Later entries overwrite earlier ones
+
+        for i, (name, stage) in enumerate(stages_by_name.items(), 1):
+            status = stage.get("status", "unknown")
             duration = (
                 stage.get("duration_ms", 0) / 1000 if stage.get("duration_ms") else None
             )
 
-            icon = (
-                "✓"
-                if status == "completed"
-                else "⏳" if status == "in_progress" else "✗"
-            )
+            # Map status to icon: completed=✓, starting/in_progress=⏳, failed=✗
+            if status == "completed":
+                icon = "✓"
+            elif status in ("starting", "in_progress"):
+                icon = "⏳"
+            elif status == "failed":
+                icon = "✗"
+            else:
+                icon = "?"  # Unknown status
             duration_str = f" ({duration:.1f}s)" if duration else ""
 
             click.echo(f"  {icon} {i}. {name}{duration_str}")

{dayhoff_tools-1.13.16 → dayhoff_tools-1.13.18}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "dayhoff-tools"
-version = "1.13.16"
+version = "1.13.18"
 description = "Common tools for all the repos at Dayhoff Labs"
 authors = [
     {name = "Daniel Martin-Alarcon", email = "dma@dayhofflabs.com"}