Commit b2fb3c3c authored by alat-rights's avatar alat-rights
Browse files

make padding optional

parent 08147d75
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -96,6 +96,7 @@ class OneHotFeaturizer(Featurizer):
      The shape is `(max_length, len(charset) + 1)`.
      The index of unknown character is `len(charset)`.
    """
    if max_length is not none:
      string = self.pad_string(string)  # Padding
    return np.array([
        one_hot_encode(val, self.charset, include_unknown_set=True)