cbds running in jupyter

materials/lib/CBDSimulatorInterpreter/cbd/examples/SimulateTrainControlCBD.py

@@ -5,7 +5,6 @@ from cbd.models.TrainCostModelBlock import StopSimulationException
 
 def simulateCBD(optimalKp, optimalKi, optimalKd):
     cbd = TrainControlCBD("train_control", optimalKp, optimalKi, optimalKd)
-    #cbd = TrainUncontrolledCBD("train_control", optimalKp, optimalKi, optimalKd)
     cbd.run(3500, 0.1)
     
     times = []

materials/lib/CBDSimulatorInterpreter/cbd/models/TrainControlCBD.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
+from cbd.models.TrainCostModelBlock import CostFunctionBlock
 from cbd.src.CBD import *
-from TrainCostModelBlock import CostFunctionBlock
 
 DELTA = 0.1
 MASS_PERSON = 73