simu_slice.py

import numpy as np
from tqdm import tqdm

from agents.round_robin_slice import RoundRobin
from associations.simple_slice import SimpleSliceAssociation
from channels.simple import SimpleChannel
from mobilities.simple import SimpleMobility
from sixg_radio_mgmt import CommunicationEnv
from traffics.simple import SimpleTraffic

seed = 10

comm_env = CommunicationEnv(
    SimpleChannel,
    SimpleTraffic,
    SimpleMobility,
    SimpleSliceAssociation,
    "simple_slice",
)

round_robin = RoundRobin(comm_env, 3, 2, np.array([8, 8]))
comm_env.set_agent_functions(
    round_robin.obs_space_format,
    round_robin.action_format,
    round_robin.calculate_reward,
)

obs = comm_env.reset(seed=seed)[0]
number_steps = 10
for step_number in tqdm(np.arange(comm_env.max_number_steps)):
    sched_decision = round_robin.step(obs)
    obs, _, end_ep, _, _ = comm_env.step(sched_decision)
    if end_ep:
        comm_env.reset()