Cleanup includes

apache · Jan 17, 2019 · a793a0d · a793a0d
1 parent aed63b7
commit a793a0d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/src/operator/image/totensor_op-inl.h b/src/operator/image/totensor_op-inl.h
@@ -26,14 +26,7 @@
 #define MXNET_OPERATOR_IMAGE_TOTENSOR_OP_INL_H_
 
 
-#include <mxnet/base.h>
-#include <algorithm>
 #include <vector>
-#include <cmath>
-#include <limits>
-#include <utility>
-#include "../mxnet_op.h"
-#include "../operator_common.h"
 #include "../elemwise_op_common.h"
 
 namespace mxnet {

diff --git a/tests/python/unittest/test_gluon_data_vision.py b/tests/python/unittest/test_gluon_data_vision.py
@@ -32,13 +32,13 @@ def test_to_tensor():
     data_in = np.random.uniform(0, 255, (300, 300, 3)).astype(dtype=np.uint8)
     out_nd = transforms.ToTensor()(nd.array(data_in, dtype='uint8'))
     assert_almost_equal(out_nd.asnumpy(), np.transpose(
-        data_in.astype(dtype=np.float32) / 255.0, (2, 0, 1)))
+                        data_in.astype(dtype=np.float32) / 255.0, (2, 0, 1)))
 
     # 4D Input
     data_in = np.random.uniform(0, 255, (5, 300, 300, 3)).astype(dtype=np.uint8)
     out_nd = transforms.ToTensor()(nd.array(data_in, dtype='uint8'))
     assert_almost_equal(out_nd.asnumpy(), np.transpose(
-        data_in.astype(dtype=np.float32) / 255.0, (0, 3, 1, 2)))
+                        data_in.astype(dtype=np.float32) / 255.0, (0, 3, 1, 2)))
 
     # Invalid Input
     invalid_data_in = nd.random.uniform(0, 255, (5, 5, 300, 300, 3)).astype(dtype=np.uint8)