Cleanup SequenceEmbedEncoder

ludwig-ai · Apr 20, 2020 · 236a83f · 236a83f
1 parent 0b8dbb1
commit 236a83f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ludwig/models/modules/sequence_encoders.py b/ludwig/models/modules/sequence_encoders.py
@@ -88,9 +88,9 @@ def __init__(
             pretrained_embeddings=None,
             embeddings_on_cpu=False,
             initializer=None,
-            reduce_output='sum',
             regularizer=None,
             dropout_rate=0,
+            reduce_output='sum',
             **kwargs
     ):
         """
@@ -210,7 +210,7 @@ def call(
 
         hidden = reduce_sequence(embedded_sequence, self.reduce_output)
 
-        return hidden  #, embedding_size # todo tf2 code cleanup
+        return hidden
 
 
 class ParallelCNN(object):