From 0c6d1cfaa7ab63f402586f71cd2f64a61d3f3754 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Thu, 19 Sep 2019 09:40:06 +0200
Subject: [PATCH] #168 Replay class with penalty
---
tests/test_utils.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/test_utils.py b/tests/test_utils.py
index 6347bd0f..96761583 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -1,5 +1,5 @@
"""Test Utils."""
-from typing import List, Tuple
+from typing import List, Tuple, Optional
from attr import attrs, attrib
@@ -13,6 +13,7 @@ class Replay(object):
direction = attrib(type=Grid4TransitionsEnum)
action = attrib(type=RailEnvActions)
malfunction = attrib(default=0, type=int)
+ penalty = attrib(default=None, type=Optional[float])
@attrs
--
GitLab