Skip to content

Commit 0ad4271

Browse files
authored
Support >=1 GPU in tests
http://b/241928108
1 parent 9d614db commit 0ad4271

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_tensorflow.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ test_that("check gpu device", {
55

66
library(tensorflow)
77
gpus = tf$config$experimental$list_physical_devices('GPU')
8-
expect_equal(length(gpus), 1)
8+
expect_gte(length(gpus), 1)
99
})
1010

1111
test_that("tensorflow with gpu", {

0 commit comments

Comments
 (0)