Skip to content

Commit 9eb0a8c

Browse files
8274 Relax gpu load check (#8282)
Related to #8274 , this PR is used to check potential issues. When I used the same environment as the nightly test, the error was not reproduced. Therefore, I hope the new change can show more information about the error. ### Description A few sentences describing the changes proposed in this pull request. ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. --------- Signed-off-by: Yiheng Wang <vennw@nvidia.com>
1 parent 996e876 commit 9eb0a8c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_load_image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ def test_nibabel_reader_gpu(self, input_param, filenames, expected_shape):
233233
input_param_cpu = input_param.copy()
234234
input_param_cpu["to_gpu"] = False
235235
result_cpu = LoadImage(image_only=True, **input_param_cpu)(filenames)
236-
self.assertTrue(torch.allclose(result_cpu, result.cpu(), atol=1e-6))
236+
assert_allclose(result_cpu, result.cpu(), atol=1e-6)
237237

238238
@parameterized.expand([TEST_CASE_6, TEST_CASE_7, TEST_CASE_8, TEST_CASE_8_1, TEST_CASE_9])
239239
def test_itk_reader(self, input_param, filenames, expected_shape):

0 commit comments

Comments
 (0)