Skip to content

Commit ef8968f

Browse files
committed
Remove numpy dependency
1 parent bb169da commit ef8968f

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

packaging/build_wheel.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ export BUILD_TYPE="wheel"
1616

1717
export SOURCE_ROOT_DIR="$PWD"
1818
setup_env
19-
pip_install numpy future wheel
19+
pip_install future wheel
2020
setup_pip_pytorch_version
2121

2222
git submodule update --init --recursive

test/test_tfrecord.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from functools import partial
1010

1111
import expecttest
12-
import numpy as np
1312

1413
import torch
1514

@@ -39,7 +38,7 @@ def setUp(self):
3938
self.temp_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "_fakedata", "tfrecord")
4039

4140
def assertArrayEqual(self, arr1, arr2):
42-
np.testing.assert_array_equal(arr1, arr2)
41+
self.assertEqual(arr1, arr2)
4342

4443
def _ground_truth_data(self):
4544
for i in range(4):

0 commit comments

Comments
 (0)