# -*- coding: utf-8 -*-

import scipy.io as sio

from app.core.config import settings

q_learning_models = {}
ahu_supply_air_temp_set_dict = {}


async def load_q_learning_model():
    base_path = settings.PROJECT_DIR
    summer_model_path = f'{base_path}/app/resources/ml_models/equipment/fcu/q_learning/net_1_summer.mat'
    winter_model_path = f'{base_path}/app/resources/ml_models/equipment/fcu/q_learning/net_1_winter.mat'
    q_learning_models.update({'summer': sio.loadmat(summer_model_path)['net'][0, 0][0]})
    q_learning_models.update({'winter': sio.loadmat(winter_model_path)['net'][0, 0][0]})