37 lines
808 B
Python
37 lines
808 B
Python
import math
|
|
from decimal import Decimal
|
|
from datetime import timedelta
|
|
import numpy as np
|
|
from typing import Dict, List
|
|
|
|
from cadCAD.configuration import append_configs
|
|
from cadCAD.configuration.utils import bound_norm_random, ep_time_step, config_sim, access_block
|
|
|
|
from .genesis_states import genesis_states
|
|
from .partial_state_update_block import partial_state_update_block
|
|
|
|
params: Dict[str, List[int]] = {
|
|
'drip_frequency': [30,60,90] # in days
|
|
}
|
|
|
|
|
|
sim_config = config_sim({
|
|
'N': 5,
|
|
'T': range(100), #day
|
|
'M': params,
|
|
})
|
|
|
|
seeds = {
|
|
'p': np.random.RandomState(26042019),
|
|
}
|
|
env_processes = {}
|
|
|
|
|
|
append_configs(
|
|
sim_configs=sim_config,
|
|
initial_state=genesis_states,
|
|
seeds=seeds,
|
|
env_processes=env_processes,
|
|
partial_state_update_blocks=partial_state_update_block
|
|
)
|