Cast to int in build_sequence_matrix::pad (#1295)

ludwig-ai · Sep 17, 2021 · 5ee9958 · 5ee9958
1 parent 3ac82ea
commit 5ee9958
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ludwig/utils/strings_utils.py b/ludwig/utils/strings_utils.py
@@ -259,7 +259,7 @@ def build_sequence_matrix(
     max_length = length_limit
 
     def pad(vector):
-        sequence = np.full((max_length,),
+        sequence = np.full((int(max_length),),
                            inverse_vocabulary[padding_symbol],
                            dtype=format_dtype)
         limit = min(vector.shape[0], max_length)