update

Signed-off-by: Yiheng Wang <[email protected]>
Project-MONAI · Jan 13, 2025 · bc9a8a5 · bc9a8a5
1 parent 1c00ea2
commit bc9a8a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_load_image.py b/tests/test_load_image.py
@@ -217,7 +217,7 @@ def test_nibabel_reader(self, input_param, filenames, expected_shape):
     @SkipIfNoModule("kvikio")
     @parameterized.expand([TEST_CASE_GPU_1, TEST_CASE_GPU_2, TEST_CASE_GPU_3, TEST_CASE_GPU_4])
     def test_nibabel_reader_gpu(self, input_param, filenames, expected_shape):
-        test_image = np.random.rand(128, 128, 128)
+        test_image = torch.randint(0, 256, (128, 128, 128), dtype=torch.uint8).numpy()
         with tempfile.TemporaryDirectory() as tempdir:
             for i, name in enumerate(filenames):
                 filenames[i] = os.path.join(tempdir, name)