test_load.py

import os
os.environ["MLFLOW_TRACKING_URI"] = "file:./mlruns"
from src.models.api import load_best_model, model
import logging
logging.basicConfig(level=logging.INFO)

print("Before load:", model)
load_best_model()

import src.models.api as api
print("After load:", api.model)