diff options
Diffstat (limited to 'tensorflow/contrib/factorization/python/kernel_tests/clustering_ops_test.py')
-rw-r--r-- | tensorflow/contrib/factorization/python/kernel_tests/clustering_ops_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/factorization/python/kernel_tests/clustering_ops_test.py b/tensorflow/contrib/factorization/python/kernel_tests/clustering_ops_test.py index 7aa58b8021..f6d035a2c6 100644 --- a/tensorflow/contrib/factorization/python/kernel_tests/clustering_ops_test.py +++ b/tensorflow/contrib/factorization/python/kernel_tests/clustering_ops_test.py @@ -125,7 +125,7 @@ class NearestCentersLargeTest(test.TestCase): # Tile points and expected results to reach requested size (num_points) (self._points, self._expected_nearest_neighbor_indices, self._expected_nearest_neighbor_squared_distances) = ( - np.tile(x, (num_points / points_per_tile, 1)) + np.tile(x, (int(num_points / points_per_tile), 1)) for x in (points, expected_nearest_neighbor_indices, expected_nearest_neighbor_squared_distances)) |