Skip to content

Commit c9d7d9c

Browse files
Update examples/keras_rs/hstu.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 79b85b6 commit c9d7d9c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/keras_rs/hstu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def create_tf_dataset(dataset_name, batch_size, max_seq_len, num_targets=1):
101101
seq_lengths = np.array([len(s) for s in item_id_sequences], dtype=np.int32)
102102

103103
print("Padding sequences using tf.keras.utils.pad_sequences...")
104-
padded_item_ids = tf.keras.utils.pad_sequences(
104+
padded_item_ids = keras.utils.pad_sequences(
105105
item_id_sequences,
106106
maxlen=max_seq_len,
107107
padding="pre",

0 commit comments

Comments
 (0)