Merge branch '168-multispeed-penalties' of...

Merge branch '168-multispeed-penalties' of gitlab.aicrowd.com:flatland/flatland into 169-Enhance_sparse_rail_generator

tests/test_utils.py

 """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