Internal change
PiperOrigin-RevId: 375727465
Change-Id: I5ea18ed73144d7d8cb5ec886c30d6250003ee13d
diff --git a/tensorflow/python/keras/datasets/imdb.py b/tensorflow/python/keras/datasets/imdb.py
index 34ecb66..ff95656 100644
--- a/tensorflow/python/keras/datasets/imdb.py
+++ b/tensorflow/python/keras/datasets/imdb.py
@@ -152,10 +152,8 @@
xs = [[w for w in x if skip_top <= w < num_words] for x in xs]
idx = len(x_train)
- x_train = np.array(xs[:idx], dtype='object')
- y_train = np.array(labels[:idx], dtype='object')
- x_test = np.array(xs[idx:], dtype='object')
- y_test = np.array(labels[idx:], dtype='object')
+ x_train, y_train = np.array(xs[:idx]), np.array(labels[:idx])
+ x_test, y_test = np.array(xs[idx:]), np.array(labels[idx:])
return (x_train, y_train), (x_test, y_test)
diff --git a/tensorflow/python/keras/datasets/reuters.py b/tensorflow/python/keras/datasets/reuters.py
index e95e05f..9c344f6 100644
--- a/tensorflow/python/keras/datasets/reuters.py
+++ b/tensorflow/python/keras/datasets/reuters.py
@@ -140,10 +140,9 @@
xs = [[w for w in x if skip_top <= w < num_words] for x in xs]
idx = int(len(xs) * (1 - test_split))
- x_train = np.array(xs[:idx], dtype='object')
- y_train = np.array(labels[:idx], dtype='object')
- x_test = np.array(xs[idx:], dtype='object')
- y_test = np.array(labels[idx:], dtype='object')
+ x_train, y_train = np.array(xs[:idx]), np.array(labels[:idx])
+ x_test, y_test = np.array(xs[idx:]), np.array(labels[idx:])
+
return (x_train, y_train), (x_test, y_test)
