diff --git a/ci/conditional-test-policy.json b/ci/conditional-test-policy.json index 9e6d2fc..0394c84 100644 --- a/ci/conditional-test-policy.json +++ b/ci/conditional-test-policy.json @@ -89,7 +89,6 @@ "obliteratus/informed_pipeline.py", "obliteratus/lora_ablation.py", "obliteratus/models/loader.py", - "obliteratus/strategies/utils.py", "obliteratus/sweep.py" ] }, diff --git a/ci/test-risk-map.json b/ci/test-risk-map.json index dd3a463..289ef64 100644 --- a/ci/test-risk-map.json +++ b/ci/test-risk-map.json @@ -380,9 +380,7 @@ "tests/test_strategies.py", "tests/test_strategy_navigation_contracts.py" ], - "conditional_gates": [ - "model-download-runtime" - ] + "conditional_gates": [] }, { "path": "obliteratus/community.py", diff --git a/tests/conditional/test_model_download_runtime.py b/tests/conditional/test_model_download_runtime.py index eb7a9b9..699de55 100644 --- a/tests/conditional/test_model_download_runtime.py +++ b/tests/conditional/test_model_download_runtime.py @@ -7,8 +7,9 @@ import uuid import pytest import torch -from obliteratus.models.loader import load_model -from transformers import AutoConfig +from obliteratus.architecture_profiles import ArchitectureClass, detect_architecture +from obliteratus.models.loader import _select_model_class, load_model +from transformers import AutoConfig, AutoModelForImageTextToText pytestmark = [pytest.mark.network, pytest.mark.download] @@ -64,3 +65,9 @@ def test_pinned_mistral4_config_resolves_composite_contract_without_remote_code( assert config.text_config.model_type == "mistral4" assert config.text_config.n_routed_experts == 128 assert config.text_config.num_experts_per_tok == 4 + assert _select_model_class("causal_lm", config) is AutoModelForImageTextToText + + profile = detect_architecture(MISTRAL4_MODEL, config=config) + assert profile.model_type == "mistral4" + assert profile.arch_class is ArchitectureClass.LARGE_MOE + assert (profile.num_experts, profile.num_active_experts) == (128, 4)