predictions.py 537 Bytes
Newer Older
u214892's avatar
u214892 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
"""
Collection of environment-specific PredictionBuilder.
"""

from flatland.core.env_prediction_builder import PredictionBuilder


class DummyPredictorForRailEnv(PredictionBuilder):
    """
    DummyPredictorForRailEnv object.

    This object returns predictions for agents in the RailEnv environment.
    The prediction acts as if no other agent is in the environment and always takes the forward action.
    """

    def __init__(self):
        pass

    def reset(self):
        pass

    def get(self, handle=0):
        return {}