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

fixed issues with editor with refactored rendertools.py

removed prints for debugging.
updated notebooks to new rendering code --> Tests should pass
parent 8d5e7ebc
No related branches found
No related tags found
No related merge requests found
......@@ -22,12 +22,12 @@ class TreeObsForRailEnv(ObservationBuilder):
For details about the features in the tree observation see the get() function.
"""
observation_dim = 9
def __init__(self, max_depth, predictor=None):
super().__init__()
self.max_depth = max_depth
self.observation_dim = 9
# Compute the size of the returned observation vector
size = 0
pow4 = 1
......
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