pagina unica

tests/unit/test_ml_registry.py

@@ -0,0 +1,23 @@
+"""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"