25 lines
1.1 KiB
Python
25 lines
1.1 KiB
Python
#from ui.config import state_dict, mechanisms, exogenous_states, env_processes, sim_config
|
|
#from ui.toyConfig import state_dict, mechanisms, exogenous_states, env_processes, sim_config
|
|
from ui.simpleBC_Config import state_dict, mechanisms, exogenous_states, env_processes, sim_config
|
|
#from ui.<config_filename> import state_dict, mechanisms, exogenous_states, env_processes, sim_config
|
|
from engine.configProcessor import generate_config
|
|
from engine.mechanismExecutor import simulation
|
|
from engine.utils import flatten
|
|
|
|
#from tabulate import tabulate
|
|
import pandas as pd
|
|
|
|
def main():
|
|
states_list = [state_dict]
|
|
configs = generate_config(mechanisms, exogenous_states)
|
|
# p = pipeline(states_list, configs, env_processes, range(10))
|
|
N = sim_config['N']
|
|
r = range(sim_config["R"])
|
|
#r = range(5)
|
|
# Dimensions: N x r x mechs
|
|
s = simulation(states_list, configs, env_processes, r, N)
|
|
result = pd.DataFrame(flatten(s))
|
|
print('Test')
|
|
# print(tabulate(result, headers='keys', tablefmt='psql'))
|
|
# remove print and tabulate functions, so it returns a dataframe
|
|
return result |