24 lines
732 B
Python
24 lines
732 B
Python
"""Tests for dataset and model YAML registries."""
|
|
|
|
from pathlib import Path
|
|
|
|
from forja_core.registry.dataset_store import FileSystemDatasetStore
|
|
from forja_core.registry.model_store import FileSystemModelStore
|
|
|
|
FIXTURES = Path(__file__).parent.parent / "fixtures"
|
|
|
|
|
|
def test_get_dataset_active_version() -> None:
|
|
store = FileSystemDatasetStore(FIXTURES / "datasets")
|
|
ds = store.get_dataset("incident_sft")
|
|
assert ds.name == "incident_sft"
|
|
assert ds.version == "v1"
|
|
assert ds.format == "jsonl"
|
|
|
|
|
|
def test_get_model_active_version() -> None:
|
|
store = FileSystemModelStore(FIXTURES / "models")
|
|
m = store.get_model("gpt4o_lora_base")
|
|
assert m.adaptation == "lora"
|
|
assert m.base_model_id == "gpt-4o"
|