fix ci test unit script

jperez999 · jperez999 · commit e00be4d6d160 · 2022-11-22T08:53:29.000-05:00
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
 
