To use InputGenerator.StopLight:

from InputGenerator.StopLight import StopLight

test_light = StopLight()

test_light.fill()

test_light.randomize()

test_light.calc_costs()

inputs = test_light.get_input()

outputs = test_light.get_output()