cleanup

Author: albertz

tests/test_nn_transformer.py

@@ -30,9 +30,7 @@ def test_nn_transformer_search():
     data, source_spatial_axis=time_dim,
     target=nn.SearchFunc(
       beam_size=3,
-      max_seq_len=nn.reduce(
-        nn.length(data, axis=time_dim),
-        mode="max", axis=nn.batch_dim)))
+      max_seq_len=nn.dim_value(data, axis=time_dim)))
   out_labels.mark_as_default_output()
 
   config_code = nn.get_returnn_config().get_complete_py_code_str(transformer)