def pad_dataset(subset, side="<STR_LIT:right>", length=-<NUM_LIT:1>):
assert length == -<NUM_LIT:1> or length > <NUM_LIT:0><EOL>if type(subset[<NUM_LIT:0>][<NUM_LIT:0>][<NUM_LIT:0>]) in [float, int, np.int64, np.int32, np.float32]:<EOL><INDENT>return _pad_2d(subset, side, length)<EOL><DEDENT>else:<EOL><INDENT>return _pad_3d(subset, side, length)<EOL><DEDENT>
Pad data set to specified length.
Parameters:
    length - max length, a just to the max length in the batch if length is -1