fix unit tests

7 jobs for 55_gpu_mode in 9 minutes and 30 seconds (queued for 1 second)
Status Job ID Name Coverage
  Prepare Env
passed #194948590
linux
prepare_env:linux

00:01:08

passed #194948591
macos
prepare_env:macos

00:02:10

 
  Test
passed #194948593
cuda-cc52 linux
functional-test:linux

00:01:50

failed #194948595
macos
functional-test:macos

00:04:43

passed #194948592
cuda-cc52 linux
unit-test:linux

00:00:42

passed #194948594
macos
unit-test:macos

00:02:05

 
  Cleanup
passed #194948596
package
cleanup

00:00:04

 
Name Stage Failure
failed
functional-test:macos Test
skipped 'cuda is not available'

----------------------------------------------------------------------
Ran 2 tests in 15.325s

OK (skipped=1)
INFO:__main__:False
Test failure !
ERROR: Job failed: exit status 1