lyceum-cli 1.0.31__py3-none-any.whl → 1.0.33__py3-none-any.whl

lyceum/external/compute/execution/gpu_selection.py

@@ -59,6 +59,22 @@ GPU_VRAM_GB = {
     "gpu.rtx6000pro": 48,
 }
 
+# Fallback pricing per hour (used when API doesn't return prices)
+GPU_PRICE_PER_HOUR = {
+    "cpu": 0.20,
+    "gpu": 0.39,
+    "gpu.t4": 0.39,
+    "gpu.t4.64gb": 1.50,
+    "gpu.a100": 2.00,
+    "gpu.a100.40gb": 1.95,
+    "gpu.a100.80gb": 1.99,
+    "gpu.h100": 3.29,
+    "gpu.h200": 3.69,
+    "gpu.l40s": 1.49,
+    "gpu.b200": 5.69,
+    "gpu.rtx6000pro": 1.95,
+}
+
 
 def format_gpu_name(profile: str) -> str:
     """Format GPU profile name for display."""
@@ -95,6 +111,9 @@ def fetch_gpu_pricing() -> dict[str, float]:
 def calculate_cost(execution_time_s: float, hardware_profile: str, pricing: dict[str, float]) -> float | None:
     """Calculate cost based on execution time and GPU pricing."""
     price_per_hour = pricing.get(hardware_profile)
+    # Fallback to hardcoded pricing if API doesn't have this profile
+    if price_per_hour is None or price_per_hour == 0:
+        price_per_hour = GPU_PRICE_PER_HOUR.get(hardware_profile)
     if price_per_hour is None or price_per_hour == 0:
         return None
     return execution_time_s * (price_per_hour / 3600)
@@ -670,14 +689,6 @@ def predict_memory(
     imports: list[str] | None = typer.Option(
         None, "--import", help="Pre-import modules (can be used multiple times)"
     ),
-    mixed_precision: str | None = typer.Option(
-        None, "--mixed-precision", "-mp",
-        help="Mixed precision dtype (fp16, bf16)"
-    ),
-    strategy: str | None = typer.Option(
-        None, "--strategy", "-s",
-        help="Parallelization strategy (ddp, fsdp, zero1, zero2, zero3)"
-    ),
     use_config: bool = typer.Option(
         True, "--use-config/--no-config",
         help="Use workspace config from .lyceum/config.json if available"
@@ -689,9 +700,7 @@ def predict_memory(
     full GPU profiling. Faster than 'predict run'.
 
     Examples:
-        lyceum predict memory train.py
-        lyceum predict memory train.py --mixed-precision fp16
-        lyceum predict memory train.py --strategy fsdp
+        lyceum gpu-selection memory train.py
     """
     status = StatusLine()
 
@@ -730,12 +739,6 @@ def predict_memory(
         if import_files:
             payload["import_files"] = import_files
 
-        # TODO: When backend supports it, add mixed_precision and strategy to payload
-        if mixed_precision:
-            console.print(f"[dim]Note: --mixed-precision {mixed_precision} (backend support coming soon)[/dim]")
-        if strategy:
-            console.print(f"[dim]Note: --strategy {strategy} (backend support coming soon)[/dim]")
-
         execution_id = submit_gpu_selection(payload, status)
         console.print(f"[dim]Execution ID: {execution_id}[/dim]")
 

lyceum/external/compute/execution/python.py

@@ -73,15 +73,22 @@ def load_workspace_config(file_path: Path | None = None) -> dict | None:
 
 def read_code_from_source(code_or_file: str, status: StatusLine = None) -> tuple[str, Path | None, str | None]:
     """Read code from file or return as-is if it's inline code."""
-    file_path = Path(code_or_file) if Path(code_or_file).exists() else None
+    # Check if input looks like a file path
+    looks_like_file = code_or_file.endswith((".py", ".ipynb")) or "/" in code_or_file or "\\" in code_or_file
 
-    if file_path:
+    file_path = Path(code_or_file)
+    if file_path.exists():
         if status:
             status.update(f"Reading {file_path.name}...")
         with open(file_path) as f:
             code_content = f.read()
         return code_content, file_path, file_path.name
 
+    # If it looks like a file but doesn't exist, raise an error
+    if looks_like_file:
+        raise FileNotFoundError(f"File not found: {code_or_file}")
+
+    # Otherwise treat as inline code
     return code_or_file, None, None
 
 

lyceum/shared/config.py

@@ -129,7 +129,7 @@ class _Config:
         if self.is_token_expired():
             console.print("[dim]Token expired, attempting refresh...[/dim]")
             if not self.refresh_access_token():
-                console.print("[red]Token refresh failed. Please run 'lyceum login' again.[/red]")
+                console.print("[red]Token refresh failed. Please run 'lyceum auth login' again.[/red]")
                 raise typer.Exit(1)
 
         # Return config instance - commands use httpx directly

{lyceum_cli-1.0.31.dist-info → lyceum_cli-1.0.33.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lyceum-cli
-Version: 1.0.31
+Version: 1.0.33
 Summary: Command-line interface for Lyceum Cloud Execution API
 Home-page: https://lyceum.technology
 Author: Lyceum Team

{lyceum_cli-1.0.31.dist-info → lyceum_cli-1.0.33.dist-info}/RECORD

@@ -8,9 +8,9 @@ lyceum/external/compute/execution/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZ
 lyceum/external/compute/execution/config.py,sha256=6JJgLJnDPTwevEaNdB1nEICih_qbBmws5u5_S9gj7k0,8866
 lyceum/external/compute/execution/docker.py,sha256=0Y6lxJAm56Jrl0HxeNz1mX6DGs556i2iMN9_U1JQP0c,9635
 lyceum/external/compute/execution/docker_compose.py,sha256=YsWPnw5nB1ZpqjU9X8o_klT78I5m46PapVwVEeWra_8,9189
-lyceum/external/compute/execution/gpu_selection.py,sha256=B6x8enjdH07eUR4s1eobxGmekLyJ8-IbadxFjOAjApw,39109
+lyceum/external/compute/execution/gpu_selection.py,sha256=LX3Ar4srw6s8PcKxASnzVOrLH6tmU1vl7rHSAkrsoEE,38900
 lyceum/external/compute/execution/notebook.py,sha256=Gw9UhJ-UjYhpjdIYQ4IMYhVjhSkAFpOQ9aFYj1qOeww,7542
-lyceum/external/compute/execution/python.py,sha256=8Y9ElWs9RdauQbhECKcBPSoT0XZeGhXZ_pkEpr3sGro,12878
+lyceum/external/compute/execution/python.py,sha256=-7GVRpQlQwn5Je_7B-LZAagEZ5mE5V4Ijs2wRThd43Q,13193
 lyceum/external/compute/execution/workloads.py,sha256=4fsRWbYGmsQMGPPIN1jUG8cG5NPG9yV26ANJ-DtaXqc,5844
 lyceum/external/compute/inference/__init__.py,sha256=4YLoUKDEzitexynJv_Q5O0w1lty8CJ6uyRxuc1LiaBw,89
 lyceum/external/compute/inference/batch.py,sha256=F4MlR2IEi1X7qUo3UDUVRWclHLLmmjqu7KZ3RkHIjNM,4349
@@ -22,15 +22,15 @@ lyceum/external/vms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lyceum/external/vms/instances.py,sha256=8DKpI8PbyZFzk5RT-IPgoMDjkf_-HC-2pJKuSFs-5BA,11007
 lyceum/external/vms/management.py,sha256=dYEkN5Qiur-SG4G5CLOk2Rbr0HW3rK1BROSp0K6KxC8,15405
 lyceum/shared/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lyceum/shared/config.py,sha256=gEXf_hEmgUq3rSxU2SY75qg32mFS6qsruox7uvQlf3I,5334
+lyceum/shared/config.py,sha256=vgmspvId3D60ESJs2LLhZ5WUglIaJ7kaehNnWB7HEuA,5339
 lyceum/shared/display.py,sha256=-VSAfoa0yivTvxRrN2RYr2Sq1x_msZqENjnkSedmbhQ,4444
 lyceum/shared/imports.py,sha256=wEG4wfVTIqJ6MBWDRAN96iGmVCb9ST2aOqSjkbvajug,11768
 lyceum/shared/streaming.py,sha256=wFb7w7fra63y8WWaIA8_E1Z6Sx_6G-0J53Zh010eZgk,9355
 lyceum_cloud_execution_api_client/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 lyceum_cloud_execution_api_client/api/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 lyceum_cloud_execution_api_client/models/__init__.py,sha256=AMlb9R9O9aNC9hvKz_8TFpEfOolYC3VtFS5JX17kYks,4888
-lyceum_cli-1.0.31.dist-info/METADATA,sha256=bGp9mqLakov3SeduGI7Fo_V5jdsHyEzh5UsisZgBhK8,1482
-lyceum_cli-1.0.31.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-lyceum_cli-1.0.31.dist-info/entry_points.txt,sha256=Oq-9wDkxVd6MHgNiUTYwXI9SGhvR3VkD7Mvk0xhiUZo,43
-lyceum_cli-1.0.31.dist-info/top_level.txt,sha256=CR7FEMloAXgLsHUR6ti3mWNcpgje27HRHSfq8doIils,41
-lyceum_cli-1.0.31.dist-info/RECORD,,
+lyceum_cli-1.0.33.dist-info/METADATA,sha256=N3bGDPySJO1veoawv0LpK7mMETe1cWr6mHPruUcha-8,1482
+lyceum_cli-1.0.33.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lyceum_cli-1.0.33.dist-info/entry_points.txt,sha256=Oq-9wDkxVd6MHgNiUTYwXI9SGhvR3VkD7Mvk0xhiUZo,43
+lyceum_cli-1.0.33.dist-info/top_level.txt,sha256=CR7FEMloAXgLsHUR6ti3mWNcpgje27HRHSfq8doIils,41
+lyceum_cli-1.0.33.dist-info/RECORD,,

{lyceum_cli-1.0.31.dist-info → lyceum_cli-1.0.33.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.10.2)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any