Skip to content
Snippets Groups Projects
Commit 435cbb87 authored by Erik Nygren's avatar Erik Nygren
Browse files

updated custom_observation_example.py to highlight how we use predictors

and render custom obvsercations
parent 074401cd
No related branches found
No related tags found
No related merge requests found
...@@ -87,6 +87,7 @@ class ShortestPathPredictorForRailEnv(PredictionBuilder): ...@@ -87,6 +87,7 @@ class ShortestPathPredictorForRailEnv(PredictionBuilder):
""" """
def __init__(self, max_depth=20): def __init__(self, max_depth=20):
# Initialize with depth 20
self.max_depth = max_depth self.max_depth = max_depth
def get(self, custom_args=None, handle=None): def get(self, custom_args=None, handle=None):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment