use crate::model::data_center::{
model::{DataCenterModel, JobType, Location, ServerType, Source},
models::{
energy_consumption::{
LinearEnergyConsumptionModel, NonLinearEnergyConsumptionModel,
SimplifiedLinearEnergyConsumptionModel,
},
energy_cost::{LinearEnergyCostModel, QuotasEnergyCostModel},
revenue_loss::MinimalDetectableDelayRevenueLossModel,
switching_cost::SwitchingCost,
},
DataCenterModelOutputSuccess,
};
use pyo3::prelude::*;
pub fn submodule(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_class::<DataCenterModel>()?;
m.add_class::<Location>()?;
m.add_class::<ServerType>()?;
m.add_class::<Source>()?;
m.add_class::<JobType>()?;
m.add_class::<LinearEnergyConsumptionModel>()?;
m.add_class::<SimplifiedLinearEnergyConsumptionModel>()?;
m.add_class::<NonLinearEnergyConsumptionModel>()?;
m.add_class::<LinearEnergyCostModel>()?;
m.add_class::<QuotasEnergyCostModel>()?;
m.add_class::<MinimalDetectableDelayRevenueLossModel>()?;
m.add_class::<SwitchingCost>()?;
m.add_class::<DataCenterModelOutputSuccess>()?;
Ok(())
}