src/tests/check_model.py

import pytest 
from src.pipeline.load_model import load_llm_model 
import os 

@pytest.mark.skipif(os.getenv("CI") == "true", reason="Skip heavy model in CI")
def test_llm_model(): 
    model , tokenizer = load_llm_model() 

    assert model is not None 
    assert tokenizer is not None