fix: support port to tf2

ludwig-ai · Feb 16, 2020 · b2673eb · b2673eb
1 parent 6689168
commit b2673eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ludwig/models/combiners.py b/ludwig/models/combiners.py
@@ -138,7 +138,7 @@ def __call__(
             feature_encodings[self.main_sequence_feature]
 
         representation = main_sequence_feature_encoding['representation']
-        representations_size = representation.shape[2].value
+        representations_size = representation.shape[2]
         representations = [representation]
 
         scope_name = 'sequence_concat_combiner'