diff --git a/examples/training_example.py b/examples/training_example.py index 521cca8c786ca28c40460da95a6379a5962705b2..1eb8d03fe3a4f927888547bf9160a9730d69e294 100644 --- a/examples/training_example.py +++ b/examples/training_example.py @@ -10,7 +10,6 @@ np.random.seed(1) # Use the complex_rail_generator to generate feasible network configurations with corresponding tasks # Training on simple small tasks is the best way to get familiar with the environment -# TreeObservation = TreeObsForRailEnv(max_depth=2, predictor=ShortestPathPredictorForRailEnv()) LocalGridObs = LocalObsForRailEnv(view_height=10, view_width=2, center=2)