freesolo-flash-dev 1.0.5__tar.gz → 1.0.6__tar.gz

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: freesolo-flash-dev
-Version: 1.0.5
+Version: 1.0.6
 Summary: Flash — managed LoRA post-training (SFT/GRPO/OPD) for Freesolo environments, driven by the `flash` CLI
 Project-URL: Homepage, https://github.com/freesolo-co/flash
 Project-URL: Repository, https://github.com/freesolo-co/flash

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/flash/cli/render.py

@@ -440,7 +440,7 @@ def gpus_table(rows: list[tuple[str, int, float | None]], tip: str) -> str:
 
 def gpu_label(spec: dict, remote: dict) -> str:
     """Human-facing GPU label. Provider metadata stays internal."""
-    return remote.get("gpu") or (spec.get("gpu") or {}).get("type", "")
+    return remote.get("allocated_gpu") or (spec.get("gpu") or {}).get("type", "")
 
 
 def runs_table(runs: list[dict]) -> str:

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/flash/providers/base.py

@@ -137,19 +137,6 @@ GPU_CLASSES: tuple[GpuClass, ...] = (
 )
 
 GPU_INFO: dict[str, GpuClass] = {g.name: g for g in GPU_CLASSES}
-LEGACY_GPU_CLASSES: tuple[GpuClass, ...] = (
-    GpuClass(
-        "RTX A6000",
-        "NVIDIA_RTX_A6000",
-        48,
-        "a6000",
-        "sm86",
-        0.49,
-        lambda_name="gpu_1x_a6000",
-        vast_name="RTX A6000",
-    ),
-)
-_GPU_INFO_ALL: dict[str, GpuClass] = {**GPU_INFO, **{g.name: g for g in LEGACY_GPU_CLASSES}}
 KNOWN = tuple(GPU_INFO)
 VALIDATED = tuple(g.name for g in GPU_CLASSES if g.validated)
 
@@ -204,7 +191,7 @@ def _alias_keys(name: str) -> set[str]:
 
 
 _ALIAS_TARGETS: dict[str, set[str]] = {}
-for _info in _GPU_INFO_ALL.values():
+for _info in GPU_INFO.values():
     for _provider_name in (_info.name, _info.vast_name, *_info.vast_aliases):
         if not _provider_name:
             continue
@@ -214,7 +201,7 @@ _ALIASES: dict[str, str] = {
     key: next(iter(targets)) for key, targets in _ALIAS_TARGETS.items() if len(targets) == 1
 }
 # exact canonical names always resolve to themselves even if a provider spelling collides with them.
-_ALIASES.update({_info.name.lower(): _info.name for _info in _GPU_INFO_ALL.values()})
+_ALIASES.update({_info.name.lower(): _info.name for _info in GPU_INFO.values()})
 # full marketing names (nvidia-smi / runpod api) and historical aliases not covered by generic rules.
 _ALIASES.update(
     {
@@ -264,11 +251,7 @@ class UnreconciledCreateError(RuntimeError):
 
 
 def canonical_gpu(name: str) -> str:
-    """Normalize a friendly GPU name to an active or retired class; raise otherwise.
-
-    Retired classes resolve for teardown/pricing compatibility but stay absent from ``GPU_INFO`` so
-    allocation, display, and provider offer matching do not select them.
-    """
+    """Normalize a friendly GPU name to a managed GPU class; raise otherwise."""
     key = (name or "").strip().lower()
     if key in _ALIASES:
         return _ALIASES[key]
@@ -276,7 +259,7 @@ def canonical_gpu(name: str) -> str:
 
 
 def get_gpu_info(name: str) -> GpuClass:
-    return _GPU_INFO_ALL[canonical_gpu(name)]
+    return GPU_INFO[canonical_gpu(name)]
 
 
 def gpu_classes_for(identity_attr: str) -> list[GpuClass]:

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/flash/schema/__init__.py

@@ -397,9 +397,9 @@ def spec_from_dict(raw: dict[str, Any], run_id: str | None = None) -> JobSpec:
         # exact_type it suggests might not be provisionable there.
         if "type" in gpu_raw and not exact_type and not gpu_provider:
             authored_type_raw = gpu_raw["type"]
-            # only an active validated class is a usable exact_type pin; an unrecognized string or a
-            # retired-but-known class (e.g. RTX A6000) must not be echoed back as an exact_type, since
-            # the exact_type validation above would then reject the very config the note suggested.
+            # only an active validated class is a usable exact_type pin; an unrecognized string must
+            # not be echoed back as an exact_type, since the exact_type validation above would then
+            # reject the very config the note suggested.
             pinnable: str | None = None
             if isinstance(authored_type_raw, str):
                 try:

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/flash/server/reconcile.py

@@ -106,6 +106,7 @@ def reconcile_run(status: runner.RunStatus, *, now: float | None = None) -> bool
     later cycle (within the window) retries once the provider invoice settles."""
     now = time.time() if now is None else now
     remote = status.remote or {}
+    spec = status.spec or {}
     # raw persisted RunStatus.remote may omit started_ts or contain a falsey value. 0.0 means an
     # unknown launch rather than the epoch; falling back to created_at prevents inflated flat-rate
     # instance billing.
@@ -126,7 +127,7 @@ def reconcile_run(status: runner.RunStatus, *, now: float | None = None) -> bool
         "runId": status.run_id,
         "realizedCostUsd": realized.realized_usd,
         "provider": realized.provider,
-        "gpu": remote.get("allocated_gpu") or remote.get("gpu"),
+        "gpu": remote.get("allocated_gpu") or (spec.get("gpu") or {}).get("type"),
         "costByResource": realized.by_resource,
         "wallSeconds": realized.wall_seconds,
         "costBasis": "realized",

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "freesolo-flash-dev"
-version = "1.0.5"
+version = "1.0.6"
 description = "Flash — managed LoRA post-training (SFT/GRPO/OPD) for Freesolo environments, driven by the `flash` CLI"
 readme = "README.md"
 # RunPod Flash supports Python 3.11-3.12 (not 3.13 yet).
@@ -116,7 +116,7 @@ artifacts = ["flash/py.typed"]
 # auto-published by .github/workflows/publish-dev.yml: a push to `dev` whose version isn't yet on
 # PyPI publishes it; pushes that don't change it find it already published and no-op. So cutting a
 # dev-channel release = bump this (and [project].version) and merge to `dev`.
-version = "1.0.5"
+version = "1.0.6"
 
 [tool.pytest.ini_options]
 pythonpath = ["."]

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/tests/test_cli_render_theme.py

@@ -88,7 +88,7 @@ def test_runs_and_status_hide_provider_names(monkeypatch) -> None:
         "run_id": "flash-1",
         "state": "done",
         "spec": {"model": "Qwen/Qwen3.5-4B", "algorithm": "grpo"},
-        "remote": {"gpu": "RTX 4090", "provider": "runpod", "flash_arm": "runpod"},
+        "remote": {"allocated_gpu": "RTX 4090", "provider": "runpod", "flash_arm": "runpod"},
     }
     runs = render.runs_table([run])
     status = render.run_status(run)
@@ -100,6 +100,48 @@ def test_runs_and_status_hide_provider_names(monkeypatch) -> None:
     assert "flash_arm" not in status.lower()
 
 
+def test_runs_and_status_prefer_allocated_gpu(monkeypatch) -> None:
+    monkeypatch.setenv("FLASH_STYLE", "1")
+    monkeypatch.setenv("NO_COLOR", "1")
+
+    allocated_run = {
+        "run_id": "flash-allocated",
+        "state": "done",
+        "spec": {
+            "model": "Qwen/Qwen3.5-4B",
+            "algorithm": "grpo",
+            "gpu": {"type": "RTX Pro 6000"},
+        },
+        "remote": {"allocated_gpu": "B200"},
+    }
+    runs = render.runs_table([allocated_run])
+    status = render.run_status(allocated_run)
+    status_panel = status.split("details", 1)[0]
+    assert "B200" in runs
+    assert "B200" in status_panel
+    assert "RTX Pro 6000" not in runs
+    assert "RTX Pro 6000" not in status_panel
+
+    spec_only_run = {
+        "run_id": "flash-spec-only",
+        "state": "done",
+        "spec": {"gpu": {"type": "RTX Pro 6000"}},
+    }
+    assert "RTX Pro 6000" in render.runs_table([spec_only_run])
+    assert "RTX Pro 6000" in render.run_status(spec_only_run)
+
+    # the removed legacy remote["gpu"] key is no longer honored: the label resolves from the
+    # authoritative allocated_gpu or the provisional spec type, never the dead key.
+    legacy_key_run = {
+        "run_id": "flash-legacy-key",
+        "state": "done",
+        "spec": {"gpu": {"type": "RTX Pro 6000"}},
+        "remote": {"gpu": "B200"},
+    }
+    assert "B200" not in render.runs_table([legacy_key_run])
+    assert "RTX Pro 6000" in render.runs_table([legacy_key_run])
+
+
 def test_color_respects_no_color(monkeypatch) -> None:
     monkeypatch.setenv("FLASH_STYLE", "1")
     monkeypatch.setenv("TERM", "xterm-256color")

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/tests/test_gpus.py

@@ -64,12 +64,14 @@ def test_vast_gpu_for_offer_accepts_h100_pcie_alias():
     assert vast_gpu_for_offer("Tesla T4", 16 * 1024) is None
 
 
-def test_retired_gpu_resolves_but_is_not_active():
-    from flash.providers.base import GPU_INFO, canonical_gpu, get_gpu_info
-
-    assert canonical_gpu("RTX A6000") == "RTX A6000"
-    assert get_gpu_info("RTX A6000").vram_gb == 48
-    assert "RTX A6000" not in GPU_INFO
+def test_removed_gpu_class_is_unmanaged():
+    # RTX A6000 was a retired class kept only to resolve legacy records; it is now fully removed, so
+    # it is not a managed class and resolves nowhere (no special retired-but-known status remains).
+    from flash.providers.base import KNOWN, UnsupportedGpuError, canonical_gpu
+
+    assert "RTX A6000" not in KNOWN
+    with pytest.raises(UnsupportedGpuError):
+        canonical_gpu("RTX A6000")
 
 
 def test_expanded_gpu_table():

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/tests/test_spec_and_validation.py

@@ -682,7 +682,7 @@ def test_gpu_constraints_reject_unknown_unsupported_or_undersized_values() -> No
         spec_from_dict(_raw(**{"gpu.provider": "aws"}))
     with pytest.raises(ConfigError, match=r"gpu\.exact_type"):
         spec_from_dict(_raw(**{"gpu.exact_type": "Tesla T4"}))
-    with pytest.raises(ConfigError, match="validated"):
+    with pytest.raises(ConfigError, match=r"unsupported gpu 'RTX A6000'"):
         spec_from_dict(_raw(**{"gpu.exact_type": "RTX A6000"}))
     with pytest.raises(ConfigError, match="requires at least"):
         spec_from_dict(
@@ -730,8 +730,8 @@ def test_gpu_type_override_warning_requires_an_authored_hint(capsys) -> None:
     assert "`flash gpus` to list valid classes" in invalid_warning
     assert 'add exact_type = "H10O"' not in invalid_warning
 
-    # a retired-but-known class canonicalizes but is not validated, so it must be treated like an
-    # unrecognized hint and never echoed back as an exact_type the schema would immediately reject.
+    # an unrecognized type hint (e.g. a removed/retired class name) must be treated like any unknown
+    # string and never echoed back as an exact_type the schema would immediately reject.
     spec_from_dict(_raw(**{"gpu.type": "RTX A6000"}))
     retired_warning = capsys.readouterr().err
     assert "[gpu] type='RTX A6000' is not an active GPU class and was ignored" in retired_warning
@@ -764,7 +764,7 @@ def test_persisted_gpu_type_is_canonicalized_and_validated() -> None:
         JobSpec.from_dict({"gpu": {"type": 1}})
     with pytest.raises(ValueError, match=r"gpu\.type: unsupported gpu 'H10O'"):
         JobSpec.from_dict({"gpu": {"type": "H10O"}})
-    with pytest.raises(ValueError, match="active validated GPU class"):
+    with pytest.raises(ValueError, match=r"unsupported gpu 'RTX A6000'"):
         JobSpec.from_dict({"gpu": {"type": "RTX A6000"}})
 
 

{freesolo_flash_dev-1.0.5 → freesolo_flash_dev-1.0.6}/uv.lock

@@ -1024,7 +1024,7 @@ wheels = [
 
 [[package]]
 name = "freesolo-flash"
-version = "1.0.5"
+version = "1.0.6"
 source = { editable = "." }
 
 [package.optional-dependencies]