Module neuralops.tests.test_part11_baseline_resolution

Part 11 Verification: Checkpoint Resolution & CLI Override.

Functions

def test_explicit_cli_checkpoint_takes_precedence()

Expand source code

def test_explicit_cli_checkpoint_takes_precedence():
    with tempfile.TemporaryDirectory() as tmpdir:
        explicit_file = Path(tmpdir) / "explicit_ckpt.pt"
        explicit_file.write_text("dummy")

        out = Path(tmpdir) / "results"
        family = "roberta"
        dataset = "sst2"
        args_parent = str(explicit_file)

        if args_parent:
            resolved = Path(args_parent)
        else:
            candidates = [out / f"{family}_{dataset}_lora_baseline" / "finetuned_model.pt"]
            resolved = next((c for c in candidates if c.exists()), candidates[0])

        assert resolved == explicit_file
        assert resolved.exists()