fixing tests

Author: varisd

neuralmonkey/decoders/autoregressive.py

@@ -6,7 +6,7 @@
 The autoregressive decoder uses the while loop to get the outputs.
 Descendants should only specify the initial state and the while loop body.
 """
-from typing import NamedTuple, Callable, Optional, Any, Dict
+from typing import NamedTuple, Callable, Optional, Any, Dict, Tuple
 
 import tensorflow as tf
 

neuralmonkey/decoders/transformer.py

@@ -4,7 +4,7 @@
 """
 # TODO make this code simpler
 # pylint: disable=too-many-lines
-from typing import Any, Callable, NamedTuple, List, Union, Tuple
+from typing import Any, NamedTuple, List, Union, Tuple
 import math
 
 import tensorflow as tf