From 435cbb87df00b1bc98e015db0795494cd6cb73a6 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Tue, 30 Jul 2019 15:43:00 -0400
Subject: [PATCH] updated custom_observation_example.py to highlight how we use
predictors and render custom obvsercations
---
flatland/envs/predictions.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/flatland/envs/predictions.py b/flatland/envs/predictions.py
index 7483ebad..88a79ea7 100644
--- a/flatland/envs/predictions.py
+++ b/flatland/envs/predictions.py
@@ -87,6 +87,7 @@ class ShortestPathPredictorForRailEnv(PredictionBuilder):
"""
def __init__(self, max_depth=20):
+ # Initialize with depth 20
self.max_depth = max_depth
def get(self, custom_args=None, handle=None):
--
GitLab