Source code for cbp.builder.line_builder
from .base_builder import BaseBuilder
[docs]class LineBuilder(BaseBuilder):
def __init__(self, num_node, d, policy, rand_seed=1):
self.num_node = num_node
super().__init__(d, policy, rand_seed)
[docs] def init_graph(self):
self.add_constrained_node()
for _ in range(self.num_node - 2):
self.add_branch(is_constrained=False)
self.add_branch(is_constrained=True)