fix ci test unit script

NVIDIA-Merlin · Nov 22, 2022 · e00be4d · e00be4d
1 parent 16d4d64
commit e00be4d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ci/test_unit.sh b/ci/test_unit.sh
@@ -18,6 +18,5 @@
 set -e
 
 # Run tests
-TF_GPU_ALLOCATOR=cuda_malloc_async pytest -rsx tests/unit tests/notebook
-
+TF_GPU_ALLOCATOR=cuda_malloc_async pytest -rsx tests/unit