def mw_no_grade1_tf_no_grade3():
#This function ensures that there are no grade 1 students on Monday,Wednesday and grade 3 students on tuesday,friday
policy_list = []
def lockdown_mw(time_step):
if time_step % 7 in [0, 2]:
return True
return False
def lockdown_tf(time_step):
if time_step % 7 in [1, 4]:
return True
return False
policy_list.append(
Lockdown_Policy.agent_lockdown('Grade', ['Grade 1'], lockdown_mw))
policy_list.append(
Lockdown_Policy.agent_lockdown('Grade', ['Grade 3'], lockdown_tf))
def event_restriction_fn(agent, event_info, current_time_step):
return False
return policy_list, event_restriction_fn
def monday_no_grade1():
#This function ensures that there are no grade 1 students on Monday
policy_list = []
def lockdown_fn(time_step):
if time_step % 7 in [0]:
return True
return False
policy_list.append(
Lockdown_Policy.agent_lockdown('Grade', ['Grade 1'], lockdown_fn))
def event_restriction_fn(agent, event_info, current_time_step):
return False
return policy_list, event_restriction_fn
def mt_no_grade2_wednesday_online():
#This function ensures that there are no grade 1 students on Monday,tuesday and wednesday is online
policy_list = []
def lockdown_mt(time_step):
if time_step % 7 in [0, 1]:
return True
return False
def lockdown_wednesday(time_step):
if time_step % 7 in [2]:
return True
return False
policy_list.append(
Lockdown_Policy.agent_lockdown('Grade', ['Grade 1'], lockdown_mt))
policy_list.append(Lockdown_Policy.full_lockdown(lockdown_wednesday))
def event_restriction_fn(agent, event_info, current_time_step):
return False
return policy_list, event_restriction_fn