nematus cell is child of grucell not orthogrucell

neuralmonkey/nn/ortho_gru_cell.py

@@ -19,7 +19,7 @@ def __call__(self, inputs, state, scope="OrthoGRUCell"):
 
 
 # Note that tensorflow does not like when the type annotations are present.
-class NematusGRUCell(OrthoGRUCell):
+class NematusGRUCell(tf.contrib.rnn.GRUCell):
     """Nematus implementation of gated recurrent unit cell.
 
     The main difference is the order in which the gating functions and linear
@@ -28,11 +28,12 @@ class NematusGRUCell(OrthoGRUCell):
     The math is equivalent, in practice there are differences due to float
     precision errors.
     """
+
     def __init__(self, rnn_size, use_state_bias=False, use_input_bias=True):
         self.use_state_bias = use_state_bias
         self.use_input_bias = use_input_bias
 
-        OrthoGRUCell.__init__(self, rnn_size)
+        tf.contrib.rnn.GRUCell.__init__(self, rnn_size)
 
     def call(self, inputs, state):
         """Gated recurrent unit (GRU) with nunits cells."""